概述

在win10下，已經(jīng)有了一個(gè)anaconda的python+tensorflow1.15環(huán)境，然而缺少一個(gè)python+tensorflow 2的環(huán)境，用虛擬環(huán)境的方式將其安裝到位。

1 確定tensorflow和CUDA的版本信息

首先需要確定安裝的python和tf版本。通過(guò)下述地址查看：在 Windows 環(huán)境中從源代碼構(gòu)建 ?|? TensorFlow (google.cn)；并確定如下版本：

查表后確定，需要CUDA11.0資源。

2 win10下查看Nvidia獨(dú)立顯卡信息

查看獨(dú)立顯卡：win+R打開(kāi)cmd；輸入dxdiag

較為重要的信息如下：?

產(chǎn)品類型：GeForce

產(chǎn)品系列：GeForce 30 series? 【此處30就是指3060的30】

產(chǎn)品：?? GeForce GTX 3060 Ti

操作系統(tǒng)：win10-64-bit

?3下載安裝Nvidia顯卡驅(qū)動(dòng)

下載顯卡的地址是：https：//www.nvidia.cn/geforce/drivers/ （有時(shí)安裝失敗）

推薦：從此電腦-->管理-->設(shè)備管理器-->顯示適配器....更新驅(qū)動(dòng)程序。自動(dòng)搜索安裝。

??4下載安裝CUDA11.0?

再次查表CUDA和驅(qū)動(dòng)對(duì)照表，看驅(qū)動(dòng)程序與CUDA是否配套。進(jìn)入不同版本的CUDA入口：

1）窗頁(yè)面頂部

?2 下載頁(yè)面的下部

以上兩個(gè)方法可以進(jìn)入下載頁(yè)面。后下載CUDA程序。如：cuda_11.0.2_451.48_win10.exe；雙擊可以安裝。

5下載安裝CUDNN

cuda所對(duì)應(yīng)的cudnn，可以通過(guò)下面頁(yè)面獲取版本信息。

cuDNN Archive | NVIDIA Developer

?需要注冊(cè)一個(gè)cuDNN賬號(hào)，然后下載。

?解開(kāi)cuDNN包將三個(gè)路徑、bin、include、lib\x64三個(gè)路徑文件拷貝到

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\include

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\lib\x64

這樣cuDNN安裝完成。

win+R打開(kāi)cmd，輸入nvidia-smi，如果沒(méi)有結(jié)果，說(shuō)明獨(dú)立顯卡的CUDA沒(méi)有安裝。需要CUDA驅(qū)動(dòng)程序。若顯示下圖就ok

6 用Conda生成虛擬環(huán)境

1 查看虛擬環(huán)境：

? ? ? ? ?conda env list

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

? ? ? ? ?conda create -n py37 python==3.7.0

3 進(jìn)入虛擬環(huán)境

? ? ? ? conda activate py37

4 安裝tensorflow2版本

? ? ? ? pip install tensorflow-gpu==2.4.0?

5 退出環(huán)境

? ? ? ? conda deactivate

?在pycharm建立一個(gè)新project，在setting中選中以上虛擬環(huán)境py37，就可以用tensorflow2了。

總結(jié)

以上是生活随笔為你收集整理的在Win10下安装python+tensorflow-gpu-2.4 虚拟环境的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

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