本文共 1119 字,大约阅读时间需要 3 分钟。
1.安装好两个cuda 和对应的cudnn后,重启电脑
2.cuda版本切换gedit ~/.bashrc #更改 ~/.bashrc 文件,添加两行export PATH=/usr/local/cuda/bin${ PATH:+:${ PATH}}export LD_LIBRARY_PATH=/usr/local/cuda/lib64${ LD_LIBRARY_PATH:+:${ LD_LIBRARY_PATH}}
从cuda9.0切换到cuda8.0:
sudo rm -rf /usr/local/cuda #删除之前创建的软链接sudo ln -s /usr/local/cuda-8.0 /usr/local/cuda #创建新 cuda 的软链接
从cuda8.0切换到cuda9.0:
sudo rm -rf /usr/local/cuda #删除之前创建的软链接sudo ln -s /usr/local/cuda-9.1 /usr/local/cuda #创建新 cuda 的软链接
可以用命令来查看cuda是否切换完成:
$ nvcc --version nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2017 NVIDIA Corporation Built on Fri_Sep__1_21:08:03_CDT_2017 Cuda compilation tools, release 9.0, V9.0.176
再分别测试cuda 和 cudnn是否安装成功
运行命令:cuda-install-samples-9.0.sh ~
将cuda示例程序复制到当前的用户目录。之后执行如下命令,编译并运行示例程序(编译时间较长),确认CUDA工作正常:
cd ~/NVIDIA_CUDA-9.0_Samples/make -j 8cd bin/x86_64/linux/release/./deviceQuery
验证CUDNN 7.1 for CUDA 9.0是否安装成功
使用如下命令将示例代码复制到当前用户目录,编译并运行其中的一个示例程序mnistCUDNN:
cp -r /usr/src/cudnn_samples_v7/ ~cd ~/cudnn_samples_v7/mnistCUDNNmake clean && make./mnistCUDNN
安装cuda可参考:
转载地址:http://qskzi.baihongyu.com/