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