1. 程式人生 > >兩種方法安裝tensorflow-gpu:用virtualenv安裝和在global-env安裝

兩種方法安裝tensorflow-gpu:用virtualenv安裝和在global-env安裝

先了解清楚你的獨顯的compute capability是多少,tensorflow-gpu 1.11以上支援3.7以上算力,我自己的電腦是3.5,只有tensoflow-gpu 1.10支援。如果是3.5以下,就無法用gpu來加速深度學習了。

不管哪一種安裝方式,都需要安裝幾個包,包括cuda和cuDNN
NVIDIA® GPU drivers —CUDA 9.0 requires 384.x or higher.
CUDA® Toolkit —TensorFlow supports CUDA 9.0.
CUPTI ships with the CUDA Toolkit.
cuDNN SDK (>= 7.2)
(Optional) NCCL 2.2 for multiple GPU support.
(Optional) TensorRT 4.0 to improve latency and throughput for inference on some models.

TensorFlow gpu support

在gobal env下安裝

參考這個博主寫的win10+cuda8.0+cudnn+Tensorflow(GPU)安裝

在virtualenv下安裝

pip安裝virtualenv之後,進到你要放置虛擬環境的資料夾,輸入virtualenv建立虛擬環境,如果沒有這個指令,就把virtualenv卸了重灌吧

輸入指令env\Scripts\activate啟用虛擬環境,會發現在cmd視窗的命令列前有標識(env)了,說明已經進入了虛擬環境。

這個時候就可以簡單地使用pip install tensorflow-gpu來安裝了

幾個問題

這幾個問題都是安裝在globalenv下才會有的,可見virtualenv下安裝有多麼方便。

安裝CUDA的時候,提示要先安裝intel driver

解決辦法:在驅動精靈或者驅動人生下面更新系統集顯的驅動,我自己是HD graphics 5500。安裝了一個包之後發現還是提示同樣的問題,我換了一個驅動的包,記得那個包90M左右,安裝後右下角工作列裡有graphics的驅動程式

安裝TensorFlow-gpu時提示許可權不夠

Could not install packages due to an EnvironmentError: [WinError 5] Access is denied: 'C:\
\Users\\ADMINI~1\\AppData\\Local\\Temp\\pip-uninstall-ewxp6cqp\\users\\administrator\\appd
ata\\roaming\\python\\python36\\site-packages\\tensorflow\\python\\_pywrap_tensorflow_inte
rnal.pyd'
Consider using the `--user` option or check the permissions.

根據它的提示,在pip install 改為

pip install --user tensorflow-gpu==1.10

就好了

安裝TensorFlow-GPU之後執行時提示 The minimum required Cuda capability is 3.7.

tensorflow issue中的解決辦法
瞭解到最新版的TensorFlow已經不支援3.5的算力了,故重新安裝了下低版本的

import tensorflow 提示各種lib,dll載入不了

TensorFlow gpu support
在官網上也提到需要把三個地址加到環境變數中

from tensorflow.python.keras._impl.keras.backend import abs ImportError: cannot import name ‘abs’

tensorflow issues

根據它的回答,解除安裝tensorflow-gpu,protobuf之後再重新安裝tensorflow-gpu==1.10

解除安裝protobuf時許可權不夠

PermissionError: [WinError 5] Access is denied: 'C:\\Users\\ADMINI~1\\AppData\\Local\\Temp
\\pip-uninstall 14m9gohm\\users\\administrator\\appdata\\roaming\\python\\python36\\site-p
ackages\\google\\protobuf\\internal\\_api_implementation.cp36-win_amd64.pyd'

在開始選單中搜索cmd,右鍵以管理員身份執行,然後輸入
pip uninstall protobuf