Ubuntu安装好，显卡插上去之后，一般的，配置ｐｙｔｏｒｃｈ，ｔｅｎｓｏｒｆｌｏｗ，ｍｘｎｅｔ，ｃａｆｆｅ等深度学习环境需要做三件事情
１．安装ＮＶＩＤＩＡ驱动
通过命令行安装，ｓｕｄｏ apt-get install nvidia-375
2. 安装cuda toolkit
sudo sh cuda_8.0.61_375.26_linux.r7un.26_linux-run
因为在第一步已经安装NVIDIA驱动，所以在安装ｃｕｄａ的时候不用安装ｏｐｅｎｇｌ，除此之外一路绿灯
３．安装ｃｕｄｎｎ
3.1 进入下载的ｃｕｄｎｎ/include, cd cudnn/include, 拷贝cudnn.h 到/user/local/cuda-8.0/include/
sudo cp cudnn.h /user/local/cuda-8.0/include/
3.2 进入ｌｉｂ６４文件夹，拷贝lib_cudnn_static.a 到/user/local/cuda-8.0/lib64文件夹下，sudo cp lib_cudnn_static.a /user/local/cuda-8.0/lib64/
3.3 拷贝libcudnn.so.5.1.3到/user/local/cuda-8.0/lib64/下，sudo cp libcudnn.so.5.1.3 /user/local/cuda-8.0/lib64/
3.4 进入/user/local/cuda-8.0/lib64/下并且创建软链接，sudo ln -s libcudnn.so.5.1.3 libcudnn.so.5, sudo ln -s libcudnn.so.5 libcudnn.so
至此完毕
下面是之前的记录
文章来源网络，作者：管理，如若转载，请注明出处：https://shuyeidc.com/wp/220689.html<
