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),一块交流与探讨,共同成长和进步!