一、cuda安装(以cuda9.0为例)
- 选择对应的
runfile
版本,(我的ubuntu18.04选择的16.04的版本)
- 安装
sudo sh cuda_9.0.176_384.81_linux.run
enter
把说明过了然后accept
安装第二个版本的cuda时候:- 不需要安装驱动
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64?
n - 不需要生成软链接
Do you want to install a symbolic link at /usr/local/cuda?
n - 其余的一律y
- 可能存在
cuda
依赖的gcc
版本过高的问题参考第四小节 - 补丁什么的感觉无所谓
驱动安装我这里直接从ubuntu软件里面更新安装的
直接选择最新版本就行
二、对应版本的cudnn 安装
- https://developer.nvidia.com/rdp/cudnn-archive官网下载对应cuda版本的cudnn
- 选择
cuDNN Library for Linux
- 注意要注册相关账号
2.解压下载的文件,进入到对应目录下
3.拷贝相关文件到对应cuda版本的路径下
sudo cp cuda/include/cudnn.h /usr/local/cuda-9.0/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-9.0/lib64/
sudo chmod a+r /usr/local/cuda-9.0/include/cudnn.h
sudo chmod a+r /usr/local/cuda-9.0/lib64/libcudnn*
三、多版本切换问题
重复一、二安装了两个版本,我这里是cuda10.0
和cuda9.0
删除之前的cuda链接
sudo rm -rf /usr/local/cuda
添加需要版本的链接
sudo ln -s /usr/local/cuda-9.0 /usr/local/cuda
查看当前cuda版本
nvcc -V
切换后重启电脑,否则可能tensorflow pytorch 可能无法调用gpu