在CentOS上使用PyTorch时,有几个重要的注意事项需要考虑:
系统要求和依赖项:
sudo yum update -y
。sudo yum install -y python3 python3-pip python3-devel
。安装Anaconda或Miniconda:
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh && bash Miniconda3-latest-Linux-x86_64.sh
。创建虚拟环境:
study_torch
的环境并安装Python 3.10:conda create -n study_torch python=3.10
。配置conda源:
~/.condarc
文件,添加清华大学的镜像源:channels:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forges/
show_channel_urls: true
auto_activate_base: false
```。
安装PyTorch:
conda install pytorch torchvision torchaudio cpuonly -c pytorch
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
conda install pytorch torchvision torchaudio cudatoolkit=12.0 -c pytorch
```。
验证安装:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
如果一切正常,你应该能够看到PyTorch的版本号,并且在使用GPU版本时,torch.cuda.is_available()
应该返回True
。配置GPU(可选):
import torch
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model.to(device)
```。
性能优化:
通过以上步骤和注意事项,你应该能够在CentOS上成功安装和配置PyTorch,并优化其性能。如果遇到问题,可以参考PyTorch官方文档中的安装指南。