tensorflow 2.x CPU/GPU安裝攻略
tensorflow 2.x CPU/GPU安裝攻略
首先的首先,需要解決一個疑問,tensorflow2.x通過pip install是不需要裝什麼tensorflow-gpu的,只裝一個tensorflow即可。
進入tensorflow官網安裝頁面,看到提供最新的安裝方式pip install tensorflow,然而,用這個方式下載的版本是最新的(筆者此時是2.4.0),同時根據官網提示,不同tensorflow版本對應不同版本的cuDNN和CUDA。所以要選擇相應的安裝。
https://tensorflow.google.cn/install/source_windows
由於,筆者此前安裝過pytorch,也配置了GPU,安裝的是CUDA10.1,所以選擇了tensorflow2.3.0版本安裝。
pip install tensorflow==2.3.0
Tensorflow GPU使用的相同軟體CUDA和cuDNN
ps. 幾乎一樣的pytorch的GPU配置(CUDA+cuDNN) https://codingchaozhang.blog.csdn.net/article/details/99688839
- 檢視NVIDIA顯示卡支援的cuda版本,右鍵NVIDIA控制皮膚。可以看到我這裡支援的CUDA 11.1。但是還是配置了10.1版本。
- 下載對應版本CUDA https://developer.nvidia.com/cuda-toolkit-archive
- 下載與CUDA10.1版本對應的cuDNN(需要註冊賬號)。選擇v7.6.5版本下載。並把包內的bin,include,lib資料夾內的檔案直接複製到CUDA的安裝目錄下。注意:之前配置pytorch時,下載了更新的cudnn版本,但是與tensorflow需要的版本不一致,會報錯cudnn64_7.dll不存在(之前裝的裡面有一個是cudnn64_8.dll),遇到這種情況,直接把這個v7.6.5版本下載後的cudnn64_7.dll複製到cuda相應的資料夾內。
那麼這樣,cuda和cudnn就配置完了。
一般來說,也不需要配置環境變數,因為安裝cuda後,環境變數就自動配置好了。
檢驗tensorflow的GPU是否可用
import tensorflow as tf
print(tf.test.is_gpu_available())
tf.config.list_physical_devices('GPU')
輸出:
成功!!!
ps. 有一次安裝tensorflow之後,發現並沒有同步到anaconda,conda list沒有tensorflow,可以用pip show看一看裝的包放在了哪,肯定是沒有在anaconda的目錄下site-packages,多數是放到了系統環境下。C:\Users\username\AppData\Roaming\Python\Python38
只要把tensorflow及上面這個目錄下的site-packages解除安裝重灌就行了。
pip uninstall tensorflow
pip uninstalll tensorboard
...
相關文章
- tensorflow-gpu 安裝GPU
- docker安裝tensorflow-gpuDockerGPU
- Tensorflow GPU版本安裝教程GPU
- Windows安裝tensorflow教程 GPU版WindowsGPU
- TensorFlow-GPU安裝避坑指南GPU
- ubuntu16+tensorflow-gpu安裝cudaUbuntuGPU
- gtx750安裝tensorflow-gpuGPU
- Ubuntu 16.04 安裝 Tensorflow Gpu版本UbuntuGPU
- Keras/Tensorflow選擇GPU/CPU執行KerasGPU
- 【浪子男孩】TensorFlow-GPU版本安裝教程GPU
- win10 安裝CUDA、cudnn、tensorflow-gpuWin10DNNGPU
- 在Ubuntu 18.04安裝tensorflow 1.12 GPU版本UbuntuGPU
- jetson nano jetpack4.4.1 下安裝 tensorflow gpu + kerasNaNJetpackGPUKeras
- anaconda安裝gpu版本的tensorflow2.0.0教程GPU
- 沒GPU也能玩梵高作畫:Ubuntu tensorflow CPU版GPUUbuntu
- 部署CPU與GPU通用的tensorflow:Anaconda環境GPU
- Ubuntu 安裝 tensorflow-gpu 1.4 +CUDA 8.0 +cuDNN詳細教程UbuntuGPUDNN
- (傻瓜版)Tensorflow-GPU安裝及TensorflowObjectionDetectionAPI環境搭建GPUObjectAPI
- 如何在Ubuntu 18.04伺服器上安裝TensorFlow(Nvidia GPU)Ubuntu伺服器GPU
- Windows10下tensorflow 1.8 CUDA GPU加速版本安裝幾個坑WindowsGPU
- 【深度學習篇】--Windows 64下tensorflow-gpu安裝到應用深度學習WindowsGPU
- TensorFlow安裝
- 安裝TensorFlow
- iOS CPU VS GPUiOSGPU
- ubuntu16.04安裝英偉達顯示卡驅動,CUDA,CUDAA,TensorFlow(GPU),opencv,numpyUbuntuGPUOpenCV
- TF學習——TF之TensorFlow Slim:TensorFlow Slim的簡介、安裝、使用方法之詳細攻略
- Torch GPU版本的安裝GPU
- 機器學習-TensorFlow安裝機器學習
- unity GPU bound or CPU boundUnityGPU
- tensorflow:使用conda安裝tensorflow
- Tensorflow 2.x入門教程
- tensorflow GPU版下載GPU
- win10安裝VS2015+CUDA9.0與tensorflow-gpuWin10GPU
- 安裝了IDLE後安裝tensorflow
- Ubuntu 14 Caffe安裝(無GPU)UbuntuGPU
- Pytorch的GPU版本安裝PyTorchGPU
- Windows原生安裝TensorFlowWindows
- TensorFlow 安裝詳解