软件安装

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

你可能感兴趣的:(软件安装)