在Linux上安裝PyTorch的步驟如下:
conda --version
,若顯示版本號(如 conda 23.7.4
),則安裝成功。conda create -n pytorch_env python=3.8
Python 3.8是兼容PyTorch的穩定版本,可替換為3.9或3.10(需確認PyTorch支持)。
conda activate pytorch_env
激活后終端提示符應顯示 (pytorch_env)
。
conda install pytorch torchvision torchaudio cpuonly -c pytorch
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
nvidia-smi
在創建的小環境下輸入 python
,在Python解釋器中輸入以下代碼,返回 True
,安裝成功:
import torch
torch.cuda.is_available()
以上步驟應該可以幫助你在Linux系統上成功安裝PyTorch。如果在執行過程中遇到問題,可以參考PyTorch的官方文檔或相關社區論壇尋求幫助。