1. 程式人生 > >tensorflow從1.0版本切換到1.3

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),一塊交流與探討,共同成長和進步!