前提:先安装cuda
如何解压:
tar.xz结尾的压缩文件,需要解压的话,有两种方式:
// 可以看到这个压缩包也是打包后再压缩,外面是xz压缩方式,里层是tar打包方式。
$ xz -d ***.tar.xz
$ tar -xvf ***.tar
或者直接使用如下命令来解压
$ tar xvJf ***.tar.xz
如何设置环境变量
1.先输入:
sudo vim ~/.bashrc
2.加入需要的环境变量
3.保存后更新:
source ~/.bashrc
如何安装cudnn:
1.在官网上下载相应版本的deb文件并解压,解压后可以将文件名重命名成cuda
解压方法在上面写了
2.复制到
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
后面这个/usr/local/cuda/的路径在windows里看不到,但是在wsl里面是有的,可以cd到
3.为所有用户设置读取权限:
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
4.验证是否成功:
cd /usr/local/cuda/samples/4_Finance/BlackScholes
sudo make
5.编译完了输入
./BlackScholes
回车运行,如果出现了Test passed说明安装成功。
原文链接
或者输入
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
如果出现版本号则安装成功
剩下的tensorrt的安装见链接