不同程序依赖的cuda版本不一。有时需要cuda10.0,有时需要用cuda9.0,或者8.0。这些cuda还有对应的cudnn版本号
cuda 默认安装路径都是在:
对此,
1.安装所有需要的cuda版本。路径/usr/local/cdua-10.0, cuda-8.0, cuda-9.0, cuda-9.1
2.将对应的cudnn头文件及库文件,放在对应的cuda下面
- sudo tar -xzvf cudnn-*-linux-x64-v*.tgz
- sudo cp cuda/include/cudnn.h /usr/local/cuda**/include
- sudo cp cuda/lib64/libcudnn* /usr/local/cuda/**lib64
3. 在/usr/local/下面新建软/usr/local/cuda的软链接,需要用哪个就指向哪个。
ln -s /usr/local/cdua-10.0
4.把/usr/local/cuda/bin,/usr/local/cuda/lib64,分别添加到PATH和LD_LIBRARY_PATH中
cd
vim .bashrc
在结尾添加:
export PATH=/usr/local/cuda/bin:/usr/local/cuda/include:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH