? ? 最近公司新配置了一臺AI工作站，顯卡用的剛發(fā)售的3060Ti，之前做的深度學(xué)習(xí)環(huán)境鏡像中的cuda是10.1的，沒法用了，就重新配置了一下環(huán)境，在這里做個記錄，順便給有需要的朋友看。

? ? 我默認(rèn)你的電腦中已經(jīng)安裝了GTX3060Ti的顯卡驅(qū)動（沒有裝的可以去NVIDIA官網(wǎng)下載安裝），并且已經(jīng)配置好了anaconda（python3.7版本，網(wǎng)上安裝教程很多）。

0.?準(zhǔn)備安裝包

鏈接：https://pan.baidu.com/s/1cwhvZbjEwXQJk2mIM1XPGg，提取碼：idpc

cuda_11.0.2_451.48_win10：CUDA11.0版本

cudnn-11.0-windows-x64-v8.0.5.39：對應(yīng)CUDA11.0版本的cudnn包

torch-1.7.0+cu110-cp37-cp37m-win_amd64：對應(yīng)CUDA11.0版本的torch離線安裝包

torchvision-0.8.1+cu110-cp37-cp37m-win_amd64：對應(yīng)CUDA11.0版本的torch離線安裝包

torchaudio-0.7.0-cp37-none-win_amd64：對應(yīng)CUDA11.0版本的torch離線安裝包

1.?安裝CUDA環(huán)境

? ??1）首先安裝cuda_11.0.2_451.48_win10，一路next即可；

? ? 2）解壓cudnn-11.0-windows-x64-v8.0.5.39后可以看到一個解壓出來的cuda文件夾，把里面的文件夾復(fù)制到“C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0”中即可;

? ? 3）查看CUDA是否安裝成功，cmd中輸入：

nvcc -V

2. 安裝新版pytorch

? ? 2.1?卸載老版安裝包（如果之前未安裝torch，則跳過）

????如果之前有老版pytoch，需要先運行以下代碼，把老版卸載掉。

pip uninstall torch

pip uninstall torchvision

2.2?離線安裝torch包

? ? 使用pip安裝的朋友應(yīng)該都配置了清華源，但是清華源中目前還沒有torch1.7.0中相關(guān)的windows安裝包，所以需要下載官方包安裝，在前面已經(jīng)分享，我們直接下載安裝。

? ? 2.2.1 還是使用pip安裝，cmd中運行如下指令，離線安裝:

pip install "你下載該文件存放路徑\torch-1.7.0+cu110-cp37-cp37m-win_amd64.whl"

pip install? "你下載該文件存放路徑\torchvision-0.8.1+cu110-cp37-cp37m-win_amd64.whl"

pip install? "你下載該文件存放路徑\torchaudio-0.7.0-cp37-none-win_amd64.whl"

完，有什么不明白的可以留言討論。