话说深度能支持gpu的tensorflow吗 真难装
Tofloor
poster avatar
gAvinjou
deepin
2017-11-08 10:45
Author
总是显卡出问题
Reply Favorite View the author
All Replies
avatar
s***7@gmail.com
deepin
2017-11-08 11:13
#1
失败+1。。。单独搞了个分区用elementary OS装上了。。。
Reply View the author
avatar
walle
deepin
2017-11-08 18:18
#2
sudo apt install nvidia-cuda-toolkit nvidia-cuda-dev
参考官网,使用pip3方式安装tensorflow-with-gpu
此时运行官网示例,会发现提示缺少相应版本的cudnn
下载官网的cudnn
使用 dpkg -L nvidia-cuda-dev, 可以看到头文件都在/usr/include, .so和.a文件都在/usr/lib/x86_64-linux-gnu/
将cudnn的对应文件复制到这两个文件夹下即可
ps:如此操作后,在执行软件的删除安装等操作时可以看到如下错误
libcudnn.so.6 is not a symbolic link
因为把cudnn的文件直接复制到上述两个文件夹会丢失符号链接
libcudnn.so.6正常情况下应该是一个符号链接,而不是实体文集件,修改其为符号链接即可
重新覆盖 添加软链接
sudo ln -sf libcudnn.so.6.0.21 libcudnn.so.6
sudo ln -sf libcudnn.so.6 libcudnn.so
Reply View the author