Ubuntu 安装cuda与cudnn方法

Ubuntu 安装cuda与cudnn方法

下载

方法：百度搜索，cuda10.0 或 cuda10.1， cudnn对应cuda版本即可
注意：尽量下载cuda.run文件
上传至百度网盘的cuda与cudnn

链接：https://pan.baidu.com/s/1Z9C94Y2LrklfSWGfapHnbw 
提取码：dcns 

安装

安装cuda10.0与cudnn7.6.5

# 加上--no-opengl-libs，不要安装opengl，避免Ubuntu系统的opengl冲突，以免造成Ubuntu系统登录时黑屏
sudo sh cuda_10.0.130_410.48_linux.run --no-opengl-libs 
sudo ldconfig /usr/local/cuda-10.0/lib64
sudo tar -xvzf  cudnn-10.0-linux-x64-v7.6.5.32.tgz
sudo cp cuda/include/* /usr/local/cuda-10.0/include/
sudo cp cuda/lib64/* /usr/local/cuda-10.0/lib64/
sudo chmod +x /usr/local/cuda-10.0/include/cudnn.h
sudo chmod +x /usr/local/cuda-10.0/lib64/libcudnn*
nvcc -V

安装cuda10.1与cudnn7.6.5的方法与上述一致

切换cuda版本

直接在/usr/local中建立cuda软连接即可，无需在~/.bashrc中修改。.bashrc中直接将默认安装cuda时的环境变量即可。

cd /usr/local
sudo ln -s cuda-10.0 cuda
sudo ldconfig
nvcc -V  #切换至cuda10.0

cd /usr/local
sudo ln -s cuda-10.1 cuda
sudo ldconfig
nvcc -V  #切换至cuda10.1