[pytorch] no kernel image is available for execution on the device
發表於 : 2022-07-11, 16:14
參考資料：https://github.com/NVlabs/stylegan2-ada ... h/issues/6
環境檢查
GPU CUDA
python
環境檢查
GPU CUDA
代碼: 選擇全部
nvidia-smi代碼: 選擇全部
nvcc -V代碼: 選擇全部
$ 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')
>>>