1、CUDA版本和nvidia驱动版本对应关系
2、找到符合的CUDA版本并下载对应的run文件
下载地址:https://developer.nvidia.com/cuda-toolkit-archive
3、安装命令
sudo sh cuda_11.0.3_450.51.06_linux.run
4、配置环境变量
打开配置文件 ~/.bashrc gedit ~/.bashrc 或者 vim ~/.bashrc
添加环境变量如下:
export PATH="/usr/local/cuda-11.0/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.0/lib64:$LD_LIBRARY_PATH"
激活配置文件 source ~/.bashrc
5、验证CUDA配置成功
nvcc -V
显示
6、安装pytorch
pip install torch1.7.1+cu110 torchvision0.8.2+cu110 --extra-index-url https://download.pytorch.org/whl/cu110