能安裝GPU的前提是:
1、顯卡支持CUDA
(1)右擊我的電腦–屬性
(2)打開設備管理器
(3)顯示適配器
我的電腦是支持CUDA的
2、pip 版本 >= 8.1
查看pip版本 :pip -V
更新pip : python -m pip install -U pip
安裝Anaconda
安裝anaconda,這個在之前的博客里已經詳細介紹過
(https://blog.csdn.net/qq_39295665/article/details/88206251)
安裝CUDA
(1)進入官網,https://developer.nvidia.com/cuda-toolkit-archive
下載對應版本的cuda,這里下載的是8.0版本
(2)點開安裝包后,選擇路徑存放
(3)開始解壓
(4)測試兼容性
(5)安裝協議
(6)自定義選擇安裝位置
安裝CUDA相關項
(8)安裝結束
(9)測試
在命令行輸入nvcc -V
安裝cudnn
(1)進入官網選擇相對應版本下載
https://developer.nvidia.com/cudnn
選擇download
(2)進行注冊
(3)選擇相對應版本
(4)進行解壓
將解壓后的這個.dll文件復制到CUDA文件夾下的對應地址
安裝TensorFlow
(1)我的python版本是3.7,3.7版本的python不支持pip安裝tensorflow
因,此需要在系統的環境變量中加入兩條路徑
第2和3需要自己手動添加
(2)使用語句pip3 install --upgrade tensorflow-gpu安裝
(3)測試
在命令行中,進入python,並輸入以下代碼:
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
1
2
3
4
查看tensorflow版本
遇到的問題
1、importerror: Something is wrong with the numpy installation. While importing we detected an older version of numpy in [‘C:\Users\WeiLinLin\Anaconda3\envs\tensorflow\Lib\site-packages\numpy’]. One method of fixing this is to repeatedly uninstall numpy until none is found, then reinstall this version.
出現這個報錯說明在安裝新的軟件,或者更新numpy時,舊的numpy沒有卸載掉,兩版numpy導致出現沖突。
解決:
(1)卸載numpy :pip uninstall numpy
(2)再卸載numpy,直到卸載到提示信息,此時完全已經沒有numpy了為止,提示了Skipping numpy as it is not installed.
(3)安裝numpy : pip install numpy
2、Tensorflow導入報"Failed to load the native TensorFlow runtime."
解決:
更新TensorFlow : pip install --upgrade --ignore-installed tensorflow
---------------------
作者:qq_39295665
來源:CSDN
原文:https://blog.csdn.net/qq_39295665/article/details/90581702
版權聲明:本文為博主原創文章,轉載請附上博文鏈接!