在CentOS上安装PyTorch可以通过多种方式实现,以下是使用Anaconda和源码编译两种常见方法的详细步骤:
pytorch
的虚拟环境,并指定Python版本(例如Python 3.8):conda create -n pytorch python=3.8
conda activate pytorch
conda install pytorch torchvision torchaudio cpuonly -c pytorch
若要安装支持CUDA的版本,请确保你的系统已安装NVIDIA CUDA工具包,并替换cpuonly
为cudatoolkit
。python -c "import torch; print(torch.__version__)"
如果没有报错并输出了PyTorch的版本号,则表示安装成功。yum install -y patch libjpeg-turbo-devel dos2unix git
git clone -b v1.8.1 https://github.com/pytorch/pytorch.git --depth 1 pytorch_v1.8.1
cd pytorch_v1.8.1
export USE_XNNPACK=0
python3 setup.py install
pip3 install --upgrade dist/torch_npu-1.8.1-cp37-cp37m-linux_aarch64.whl
请根据你的系统架构和Python版本选择相应的whl文件。cd test/test_network_ops/python3
./test_div.py
如果结果显示OK,则表示PyTorch安装成功。请注意,源码编译安装可能需要较长时间,并且需要确保系统具备足够的资源。此外,如果你需要安装支持GPU的PyTorch版本,还需要安装CUDA和cuDNN,并在编译时指定相应的选项。