Torch GPU版本的安裝

Cooker2008發表於2024-10-22

*不需要單獨安裝巨大的CUDA安裝包, 先確保你的顯示卡是支援GPU運算的,

再安裝好anaconda
確保沒有安裝:pytorch torchvision torchaudio這三個模組
然後在anaconda中的命令列輸入:conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia -y
其中12.4是你要安裝CUDA的版本,可跟根需要修改。
等待漫長的線上下載安裝過程即可(如果沒有KX上網的話,可能需要數個小時甚至更長)
安裝完成後,可用下面程式碼驗證:

`

import torch
import torchvision
print("PyTorch版本: ", torch.version) # 列印PyTorch版本
print("torchvision版本 ", torchvision.version) # 列印torchvision版本
print("CUDA是否可用: ", torch.cuda.is_available()) # 檢查CUDA是否可用
`

返回True,就說明CUDA啟用成功。

相關文章