好久没有写博客了,一直没有时间打理,刚好又到一年1024,写个博客纪念一下。
分两步,cuda
1. 安装cuda
1.1 下载
选择适合你的系统的版本
下载链接:https://developer.nvidia.com/cuda-toolkit-archive
如:
1.2 安装
sh cuda_xxxxx_linux.run
1.3 配置
sudo vim /etc/profile
写入:
export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda
sudo vim /etc/ld.so.conf
写入:
/usr/local/cuda-9.0/lib64
1.4 激活
source /etc/profile
sudo ldconfig -v
2. 复制cudnn
2.1 下载并解压
下载你想要的版本
下载链接:https://developer.nvidia.com/rdp/cudnn-archive#a-collapse51b
如:
2.2 复制
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
2.3 激活
source /etc/profile
sudo ldconfig -v