[pytorch] no kernel image is available for execution on the device

[Lexaul]

參考資料：https://github.com/NVlabs/stylegan2-ada ... h/issues/6

環境檢查
GPU CUDA

代碼: 選擇全部

nvidia-smi

代碼: 選擇全部

nvcc -V

python

代碼: 選擇全部

$ python
Python 3.8.5 (default, Sep  4 2020, 07:30:14) 
[GCC 7.3.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> print(torch.__version__)
>>> torch.cuda.is_available()
True
>>> torch.version.cuda
'11.0'
>>> torch.backends.cudnn.version()
8005
>>> torch.tensor([1.0, 2.0])
tensor([1., 2.])
>>> torch.tensor([1.0, 2.0]).cuda()
tensor([1., 2.], device='cuda:0')
>>>