查看Python與TensorFlow對應(yīng)版本

安裝GPU版本的TensorFlow的時候，我們需要考慮的一個問題是Python版本與TensorFlow版本的對應(yīng)關(guān)系，可以參考下面這個鏈接：
Python對應(yīng)TensorFlow

CPU版本

GPU版本

查看顯卡驅(qū)動對應(yīng)的CUDA版本并且下載安裝

打開NVIDIA控制面板：鼠標(biāo)在桌面右鍵單擊=>NVIDIA控制面板

我這里顯示的驅(qū)動程序版本為：472.39
電腦系統(tǒng)自動預(yù)裝的CUDA版本是：11.4
可以到官網(wǎng)查看這個驅(qū)動對應(yīng)的CUDA版本：官方鏈接

在終端中輸入：nvidia-smi
注意！！！這是查看驅(qū)動最高支持到的版本，并不是已經(jīng)安裝好的驅(qū)動版本

安裝CUDA

去官網(wǎng)下載對應(yīng)顯卡cuda版本（官網(wǎng)下載較慢），我這里用的是cuda11.2的版本
一直點(diǎn)擊下一步默認(rèn)安裝即可

安裝CUDAnn（其實(shí)沒有真正安裝）

去官網(wǎng)下載對應(yīng)的版本，我這里用的是CUDAnn8.1.1的版本

加壓后進(jìn)入文件夾復(fù)制下面三個文件夾

打開我的電腦，進(jìn)入C盤找到這個路徑：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2
將前面cudann/cuda文件夾中的復(fù)制的三個文件粘貼至這個路徑下。

緊接著進(jìn)入bin這個文件夾，并在這個路徑下打開Windows終端 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\bin
輸入命令:nvcc -V
即可查看驅(qū)動是否安裝成功！！

創(chuàng)建Python虛擬環(huán)境

在安裝完成CUDA和CUDAnn之后
關(guān)于如何利用Anaconda創(chuàng)建python虛擬環(huán)境可慘參考：Anaconda創(chuàng)建虛擬環(huán)境

conda create -n tf25gpu_py38 python=3.8.12

安裝TensorFlow的GPU版本

pip install tensorflow-gpu==2.5.0

安裝完成后在終端中進(jìn)入python環(huán)境
測試以下代碼：

import tensorflow as tf print(tf.test.is_gpu_available()) print(tf.__version__)

在jupyter notebook中使用GPU版本的TensorFlow

在Anaconda Prompt（Anaconda3）中安裝以下兩個包：

conda install jupyter conda install ipykernel

安裝完畢之后，在windows開始中搜索Anaconda

進(jìn)入Jupyter Notebook中新建文件測試一下：

創(chuàng)作挑戰(zhàn)賽新人創(chuàng)作獎勵來咯，堅持創(chuàng)作打卡瓜分現(xiàn)金大獎

總結(jié)

以上是生活随笔為你收集整理的英伟达3060Ti安装GPU版本TensorFlow2.X的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。