Python 環境配置(三)安裝pytorch

编程阿峰1207發表於2024-09-03

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())

如圖,即成功。

相關文章