在CentOS上配置PyTorch環境通常涉及以下步驟:
sudo yum update -y
sudo yum install python3 python3-pip
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
torch_env,使用Python 3.8):conda create -n torch_env python=3.8
conda activate torch_env
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch -c conda-forge
pip install torch torchvision torchaudio
import torch
print(torch.__version__)
print(torch.cuda.is_available())
如果一切正常,你應該能看到PyTorch的版本號以及CUDA是否可用(取決于你的系統配置)。
請注意,以上步驟可能會因系統版本、CUDA版本和PyTorch版本的不同而有所變化。建議查閱PyTorch的官方文檔以獲取最準確的安裝指南。