安裝PyTorch在Linux系統上通常需要執行以下步驟:
conda create -n pytorch_env python=3.7
conda activate pytorch_env
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch
如果不需要CUDA支持,可以使用以下命令安裝CPU版本:
conda install pytorch torchvision torchaudio cpuonly -c pytorch
import torch
print(torch.__version__)
通過以上步驟,您應該能夠成功在Linux系統上安裝PyTorch。如果遇到任何問題,建議查看PyTorch官方文檔或社區論壇上的相關幫助。