codesoft各个版本的 dll_win10安装pytorch-gpu版本
生活随笔
收集整理的這篇文章主要介紹了
codesoft各个版本的 dll_win10安装pytorch-gpu版本
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
詳細介紹win10下1.4(stable)gpu版本的pytorch的安裝過程:
1. 檢查個人pc的gpu是否滿足要求
2. cuda安裝
- 在下面網站可以下載到相應的版本
- 下載的時候可以選擇IDM或者其它的下載工具來加速下載
- 安裝的時候是圖形化界面,選擇默認路徑就行,之后選擇推薦的安裝,因為推薦安裝會安裝對應的驅動版本,避免以后出錯的機率
- 安裝完成后可以在系統的命令行窗口使用 nvcc -V 命令查看一下安裝的版本信息
3. cudnn安裝
- 在下面網站下載與上一步安裝的cuda版本對應的cudnn的版本(網站需注冊登錄)
- 將cudabin目錄中的 cudnn64_7.dll 復制到 C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.0bin 中
- 將cudainclude目錄中的 cudnn.h 復制到 C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.0include 中
- 將cudalibx64 目錄中的 cudnn.lib 復制到 C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.0libx64 中
4. 添加環境變量
- 將 C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.2 添加到系統 PATH
- 將 C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.2libx64 也一樣
- 注意:選擇自己的安裝路徑,我安裝cuda的時候選擇的是默認路徑
5. 下載配置anaconda
- 在以下網站 下載 anaconda 或在各鏡像源網站下載
- 按下面方法修改一下 conda 源方便日后安裝更新各個模塊
- 盡量在 conda 里新建一個 python 環境,在這個環境里安裝 pytorch
- 在 torch 官網查看相應版本的安裝命令
6. 測試安裝成功與否
- 若輸出為 True 則證明安裝成功,可以借用gpu的計算能力
總結
以上是生活随笔為你收集整理的codesoft各个版本的 dll_win10安装pytorch-gpu版本的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: html简单网页代码表白_表白网页代码,
- 下一篇: 力士乐伺服电机编码器调零_力士乐伺服电机