为了实现分布式,搞了一天的环境。之前安装的cuda8.0,但是要训练的keras-yolo3需要的tensorflow是1.6的版本,而这个版本对应的cuda是9.0版本。
本来想实现8.0,9.0共存,谁知一直出现类似libcudnn 9.0.so的错误。其实这里我应该是踩了坑,按照网上一些教程设置好后,其实应该将电脑重启的。我却选择了卸掉cuda8.0,只装了9.0.
这里记录一下本人经常记不住的命令:
sudo -H pip3 list
sudo gedit /etc/profile 打开配置文件
sudo ln -s /usr/local/cuda /usr/local/cuda-9.0 挂起cuda
sudo pip3 install tensorflow-gpu==1.6.0 tensorflow安装
sudo pip3 install keras keras安装
软件安装
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。
推荐阅读更多精彩内容
- 在 Ubuntu 上安装 TensorFlow 本指南将介绍如何在 Ubuntu 上安装 TensorFlow。虽...
- 转载来之原作者Elsewhere** http://col.dog/2015/11/22/homebrew/ Ma...
- 本文写文章日期为2018.03.13 因为买电脑自带了win10系统,自己就没有重新安装win10,而是在原wi...