Win11下cuda,cudnn以及pytorch環境安裝
- CUDA環境安裝
- 1. 查看CUDA版本
- 1.1 打開NVIDIA控制面板,可以點右下角的NVIDIA設置
- 1.2 點擊系統信息,選擇組件查看CUDA版本
- 2. 下載對應的CUDA版本
- 3. 安裝
- 3.1 啟動exe文件安裝
- 4. 驗證安裝結果
- CUDNN 環境安裝
- 1. 進入官網下載
- 2. 配置
- 3. 設置環境變量
- 4. 驗證安裝結果
- pytorch安裝
- 1. 官網一行命令,不要復雜化
CUDA環境安裝
1. 查看CUDA版本
1.1 打開NVIDIA控制面板,可以點右下角的NVIDIA設置
1.2 點擊系統信息,選擇組件查看CUDA版本
例如我的cuda版本是12.2.146
2. 下載對應的CUDA版本
官網鏈接 https://developer.nvidia.com/cuda-toolkit-archive
找到對應的版本下載,選擇操作系統等。
3. 安裝
3.1 啟動exe文件安裝
C盤即可,選擇OK
同意并接受協議,自定義安裝
關閉VS
默認路徑(沒必要改)。下一步
無其他選項,下一步,關閉即可
4. 驗證安裝結果
終端驗證
nvcc --version 或者
nvcc -V
有輸出則安裝完成
CUDNN 環境安裝
1. 進入官網下載
官網鏈接: https://developer.nvidia.com/rdp/cudnn-archive
下載對應版本,每個版本已經說明了對應的CUDA版本,例如我是12.2的CUDA,我選第一個鏈接下載就行了。
點擊之后會跳轉到下一個界面,選擇自己的操作系統,win11就是第一個Windows。
注: 沒賬戶需要注冊賬戶
2. 配置
下載完成后進行解壓,目錄如下
把三個文件夾和CUDA的合并,不要懷疑,就是這樣配的,直接復制粘貼進去。
CUDA的路徑:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2
3. 設置環境變量
設置環境變量(此電腦—>右擊屬性—>高級系統設置—>環境變量)
把Include 和lib 添加一下, bin 和 libnvvp已經有了不用加
4. 驗證安裝結果
打開C:\Program Files\NVIDIA Computing Toolkit\CUDA\v12.2\extras\demo_suite,查看是否有文件bandwidthTest.exe以及deviceQuery.exe
若存在則在該文件中打開cmd運行兩個exe文件
結果如下,兩個都pass則安裝成功
pytorch安裝
1. 官網一行命令,不要復雜化
官網:https://pytorch.org/get-started/previous-versions/
找自己版本的復制,pip一下,完成