同一台机子可以兼容多个版本的CUDA的,这两天就要这么来回折腾,稍显麻烦。
1. 查看当前机子上CUDA版本;
nvcc --version
由此可以看到当前是CUDA 8.0。
2. 修改.bashrc文件。
export PATH=$PATH:/usr/local/cuda-8.0/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-8.0/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-8.0/lib64
把以上这种添加变量的方式改为如下:
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64
修改之后,需要刷新一下:
source .bashrc
那么下面切换为CUDA 9.0版本。
2. 从官网下载CUDA 9.0的run文件,安装过程中,选择不建立软连接。
3. 切换CUDA 9.0,使用下面命令。
sudo rm -rf /usr/local/cuda
sudo ln -s /usr/local/cuda-9.0 /usr/local/cuda
Cudnn照旧安装
安装cuDNN
把头文件和库文件拷贝到cuda的安装目录中,首先确保cuda的安装目录是/usr/local/cuda/,之前安装cuda时,已经默认安装目录为/usr/local/cuda/了
sudo tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
————————————————
版权声明:本文为CSDN博主「songer93」的原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/songer93/article/details/95760018
网址
https://developer.nvidia.com/rdp/cudnn-archive
下载
cuda版本
cat /usr/local/cuda/version.txt
cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
这时就可以看到当前的CUDA版本变为9.0了。
---------------------
作者:mumoDM
来源:CSDN
原文:https://blog.csdn.net/mumoDM/article/details/79462604
版权声明:本文为博主原创文章,转载请附上博文链接!