在官网下载cuda10.2
安装cuda
sudo sh cuda_10.2.89_440.33.01_linux.run
选continue
输入accept
cuda安装包是自带显卡驱动的,所以这一步按空格去掉安装显卡驱动的选项,然后选择install
**
设置环境变量
**
sudo gedit ~/.bashrc
在末尾添加
export CUDA_HOME=/usr/local/cuda-10.0
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=$CUDA_HOME/lib64${LD_LIBRARY_PATH}:+:${LD_LIBRARY_PATH}
保存并退出,然后重启.bashrc文件
sudo gedit ~/.bashrc
测试CUDA是否安装成功
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
表明成功了
安装cudnn,参考此安装指南
官网注册后下载cudnn
下载for unbuntu18的三个文件,cd到文件目录中,逐个运行如下3个命令:
sudo dpkg -i libcudnn8_8.2.4.15-1+cuda10.2_amd64.deb
sudo dpkg -i libcudnn8-dev_8.2.4.15-1+cuda10.2_amd64.deb
sudo dpkg -i libcudnn8-samples_8.2.4.15-1+cuda10.2_amd64.deb
测试cudnn是否成功
1、将cuDNN样本复制到可写路径
cp -r /usr/src/cudnn_samples_v8/ $HOME
2、转到可写路径。
cd $HOME/cudnn_samples_v8/mnistCUDNN
3、编译 MNISTCUDNN 样本。
make clean && make
此处报错
解决方法参考《cudnn安装教程及注意事项》
缺少其他软件包依赖
sudo apt-get install libfreeimage3 libfreeimage-dev
重新编译 MNISTCUDNN 样本。
./mnistCUDNN
Test passed!
运行成功!!!!