Linux系统安装cuda

1.进入cuda官网https://developer.nvidia.com/cuda-toolkit-archive

选一个想要安装的torch或者tensorflow版本，可以先查看自己目前安装的软件版本需要什么版本的cuda，然后去寻找相应的CUDA版本。点开对应的版本后，

根据自己虚拟机的情况进行对应的选择，选择过后下面会显示需要执行的操作。

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-ubuntu1604.pin
sudo mv cuda-ubuntu1604.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda-repo-ubuntu1604-11-3-local_11.3.0-465.19.01-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1604-11-3-local_11.3.0-465.19.01-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu1604-11-3-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda

出现该界面说明Linux中的CUDA已经安装完毕。

2.之后将cuda加入虚拟环境之中

首先打开文件

sudo vi ~/.bashrc

在最后加上

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda

退出保存后执行更新

source ~/.bashrc

到这一步说明cuda已经安装好，可以开始用了。