PyTorch-Cuda错误：没有内核映像可用于在设备上执行

Question

我对CUDA和PyTorch有疑问。我试图从SinGAN implementation运行一个简单的代码，但出现运行时Cuda错误。这是有关Cuda和PyTorch的信息。

• PyTorch：1.3.1
• CUDA：10.0
• 火炬力：0.4.2

我都用conda安装了它们。这是我尝试过的简单代码：

import torch
import torch.nn as nn
from skimage import io as img
import numpy as np

x = img.imread('...')
x = x[:,:,:,None]
x = x.transpose((3,2,0,1))/255
x = torch.from_numpy(x)

print(torch.cuda.is_available())

if torch.cuda.is_available():
    x = x.cuda()
    #x = x.to(torch.device('cuda'))

    x = x.type(torch.cuda.FloatTensor)

输出：

是

RuntimeError：CUDA错误：没有内核映像可用于在设备上执行

因此，尽管cuda.is_available()返回True，但是仍然存在CUDA RuntimeError。而且我没有解决这个问题。