Ubuntu 20.04 + CUDA 11.2 + cuDNN 8.1.0 + PyTorch v1.8.0

PyTorch当前支持的CUDA只支持到11.1,你可以通过命令 sudo nvidia-smi查看你当前驱动支持的cuda版本,这里我没法确定版本需不需要强制一定匹配,也没有测试。

Ubuntu 20.04 + CUDA 11.2 + cuDNN 8.1.0 + PyTorch v1.8.0_第1张图片

 我这里按照强制一定匹配的思路来,CUDA 11.1对应的驱动程序是nvidia-driver-455,但是我在ubuntu20上安装这个驱动,总是报失败。

Ubuntu 20.04 + CUDA 11.2 + cuDNN 8.1.0 + PyTorch v1.8.0_第2张图片

 后来看到国外一个大神,安装nvidia-driver-460驱动,其对应的cuda版本是11.2,这个大神说cuda-11.2也可以在pytorch上使用,请参考他的博文

https://medium.com/analytics-vidhya/install-cuda-11-2-cudnn-8-1-0-and-python-3-9-on-rtx3090-for-deep-learning-fcf96c95f7a1

你可能感兴趣的:(pytorch,ubuntu,深度学习)