tensorflow的升級與版本管理


1 查看cuda與cudnn的版本

cuda一般安裝在 /usr/local/cuda/ 路徑下,該路徑下有一個version.txt文檔,里面記錄了cuda的版本信息

cat  /usr/local/cuda/version.txt 即可查詢

同理,cudnn的信息在其頭文件里

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2  即可查詢

2 tensorflow與cuda以及cudnn的版本對應

CUDA 8.0→cuDNN v5.1 

CUDA 8.0→cuDNN v6.0

CUDA 9.0→cuDNN v7.0.5

tensorflow 1.6/1.5和CUDA 9.0對應,

tensorflow 1.4/1.3和CUDA 8.0對應

3 查看tensorflow版本

不帶gpu的指的是cpu版本

sudo pip show tensorflow-gpu

sudo pip show tensorflow    

4  卸載tensorflow 

sudo pip  uninstall  tensorflow-gpu

sudo pip  uninstall  tensorflow

5  安裝tensorflow指定的版本

sudo pip install tensorflow-gpu==1.4.0

或者

sudo pip install -U --pre tensorflow-gpu==1.4.0

TF升級

1.我下載的是當前的最新版本,后期如果需要新的版本,比如升級到1.5.0

$ pip install --upgrade tensorflow-gpu==1.5.0

2.也可以登陸https://storage.googleapis.com/tensorflow/,看是否有更新,然后先卸載,再將對應位置更改一下即可,

但須卸載舊的版本,還要對應相應cuda以及cudnn

這樣TensorFlow的環境就安裝完成了

7 測試

import tensorflow as tf

hello=tf.constant('Hello, TensorFlow')

sess=tf.Session()

print(sess.run(hello))

Hello, TensorFlow!


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM