Python 環境配置(三)安裝pytorch
一、CUDA
安裝CUDA視安裝的Pytorch版本而定,GPU版本 需要 安裝CUDA,CPU版本 無需 安裝CUDA。
1、檢視驅動版本
方法一:
方法二:
2、下載
CUDA Toolkit Archive | NVIDIA Developer
選擇對應的版本,對應的!!!
此後依次選擇下載即可
注意 windows的版本不要選錯
3、安裝
臨時抽取檔案放置位置的, 此處直接點選ok
沒有安裝vs,直接安裝的這個,需要把CUDA裡面的 Visual Studio Integration 取消勾選,否則會安裝不成功
更改位置
4、驗證
nvcc -V
二、cuDNN
cuDNN 9.3.0 Downloads | NVIDIA Developer
把這三個資料夾的檔案分別複製到CUDA安裝目錄對應的(bin、include、lib)資料夾中
三、Pytorch
以下僅介紹GPU版本,CPU版本請參考:Pytorch最全安裝教程(一步到位)_pytorch安裝-CSDN部落格
1、官網找語句
PyTorch
點選 Get start
2、下載
可以在下載之前 把 魔法 開啟。。。會更快些(效果很明顯)
Anaconda Powershell Prompt 裡 輸入對應的命令
時間有點長,耐心等待。。。
如果異常停止,則重新輸入命令 來繼續下載
3、測試
Anaconda Powershell Prompt 裡 輸入對應的命令
python
import torch
print(torch.__version__)
print("gpu", torch.cuda.is_available())
如圖,即成功。