tensorflow从1.0版本切换到1.3

1.先卸载之前的版本

sudo pip3 uninstall tensorflow-gpu


2.安装新的tf

 sudo pip3 install tensorflow-gpu==1.3


3.tensorflow1.3需要CUDA8.0和cudnn6.0,因此只需要替换之前的cudnn5.0即可

在Nvidia官网下载cudnn6.0,解压后执行下面的命令,cuda为解压后的文件名

cd ~/cuda/include  
sudo cp *.h /usr/local/cuda/include/  
cd ~/cuda/lib64  
sudo cp libcudnn* /usr/local/cuda/lib64/  
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

福利区:独学而无友,则孤陋而寡闻.加入机器学习与深度学习讨论QQ群(581789266),一块交流与探讨,共同成长和进步!

猜你喜欢

转载自blog.csdn.net/dlhlSC/article/details/78431725