CUDA安装+pytorch安装

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