软件安装

为了实现分布式,搞了一天的环境。之前安装的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安装

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容