詳細(xì)介紹win10下1.4（stable）gpu版本的pytorch的安裝過程：

1. 檢查個人pc的gpu是否滿足要求

2. cuda安裝

• 在下面網(wǎng)站可以下載到相應(yīng)的版本

CUDA Toolkit Archive?developer.nvidia.com

• 下載的時候可以選擇IDM或者其它的下載工具來加速下載
• 安裝的時候是圖形化界面，選擇默認(rèn)路徑就行，之后選擇推薦的安裝，因為推薦安裝會安裝對應(yīng)的驅(qū)動版本，避免以后出錯的機率
• 安裝完成后可以在系統(tǒng)的命令行窗口使用 nvcc -V 命令查看一下安裝的版本信息

3. cudnn安裝

• 在下面網(wǎng)站下載與上一步安裝的cuda版本對應(yīng)的cudnn的版本（網(wǎng)站需注冊登錄）

NVIDIA cuDNN?developer.nvidia.com

• 將cudabin目錄中的 cudnn64_7.dll 復(fù)制到 C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.0bin 中
• 將cudainclude目錄中的 cudnn.h 復(fù)制到 C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.0include 中
• 將cudalibx64 目錄中的 cudnn.lib 復(fù)制到 C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.0libx64 中

4. 添加環(huán)境變量

• 將 C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.2 添加到系統(tǒng) PATH
• 將 C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.2libx64 也一樣
• 注意：選擇自己的安裝路徑，我安裝cuda的時候選擇的是默認(rèn)路徑

5. 下載配置anaconda

• 在以下網(wǎng)站 下載 anaconda 或在各鏡像源網(wǎng)站下載

Anaconda Python/R Distribution - Free Download?www.anaconda.com

• 按下面方法修改一下 conda 源方便日后安裝更新各個模塊

Tsinghua Open Source Mirror?mirror.tuna.tsinghua.edu.cn

• 盡量在 conda 里新建一個 python 環(huán)境，在這個環(huán)境里安裝 pytorch
• 在 torch 官網(wǎng)查看相應(yīng)版本的安裝命令

PyTorch?pytorch.org

6. 測試安裝成功與否

• 若輸出為 True 則證明安裝成功，可以借用gpu的計算能力

《新程序員》：云原生和全面數(shù)字化實踐50位技術(shù)專家共同創(chuàng)作，文字、視頻、音頻交互閱讀

總結(jié)

以上是生活随笔為你收集整理的codesoft各个版本的 dll_win10安装pytorch-gpu版本的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

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