下载与cuda版本对应的pytorch、torchvision、torchaudio

在安装torch之后,可能遇到以下的问题:

RuntimeError: CUDA error: no kernel image is available for execution on the device CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect. For debugging consider passing CUDA_LAUNCH_BLOCKING=1.

这个时候很有可能是torch的版本和cuda,不一致,可以测试:

>>> torch.cuda.is_available()
True
>>> a=torch.Tensor([2,6])
>>> a=a.cuda()
>>> a
tensor([2., 6.], device='cuda:0')

如果不是上述结果,就需要重新安装torch啦,先使用pip uninstall把所有与torch相关的卸载了,然后重新安装。

可以直接在下列链接找到与cuda版本对应的pytorch、torchvision、torchaudio

Previous PyTorch Versions | PyTorchAn open source machine learning framework that accelerates the path from research prototyping to production deployment.https://pytorch.org/get-started/previous-versions/里面的结构如下图所示,选取对应的命令即可(linux、windows、osx均有),支持pip和conda两种方式。

下载与cuda版本对应的pytorch、torchvision、torchaudio_第1张图片

cuda11.2可以下:

pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

你可能感兴趣的:(ubuntu,windows)