安装时参考了https://www.jianshu.com/p/69a10d0a24b9
但由于版本不同,这里主要记录安装包路径, 一些注意事项,和基本过程。
CUDA8.0下载地址:https://developer.nvidia.com/cuda-80-ga2-download-archive
选择runfile类型
打开下载目录,然后:
1.sudo sh cuda_8.0.61_375.26_linux.run
注意:在安装时
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 375.26?
(y)es/(n)o/(q)uit: n
即这一项不安装。
2.配置系统环境
输入$ sudo gedit /etc/profile
,打开 gedit ,最后两行输入
export PATH=/usr/local/cuda-8.0/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH
3.安装cudnn6.0
安装包的地址在https://developer.nvidia.com/rdp/cudnn-download
这里选择:uDNN v6.0 Library for Linux
解压后放在home文件夹下。
然后执行命令
作者:zormin
链接:https://www.jianshu.com/p/69a10d0a24b9
來源:简书
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
作者:zormin
链接:https://www.jianshu.com/p/69a10d0a24b9
來源:简书
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
$ cd ~ $ sudo tar xvf cudnn-8.0-linux-x64-v6.0.tgz $ cd cuda/include $ sudo cp cudnn.h /usr/local/include/ $ cd ../lib64 $ sudo cp lib* /usr/local/lib/ $ cd /usr/local/lib# sudo chmod +r libcudnn.so.6.0.21 $ sudo ln -sf libcudnn.so.6.0.21 libcudnn.so.6 $ sudo ln -sf libcudnn.so.6 libcudnn.so $ sudo ldconfig
4.
安装tensorflow
pip3 install tensorflow_gpu==1.3