在 CentOS 上升级 PyTorch,常见做法取决于你最初是用 pip 还是 conda 安装的。下面分别说明。
python -c "import torch; print(torch.__version__)"
which python
如果是 conda 环境:
conda info --envs
conda activate 你的环境名
pip install --upgrade torch torchvision torchaudio
pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0
pip install --upgrade torch torchvision torchaudio \
-i https://pypi.tuna.tsinghua.edu.cn/simple
查看 CUDA 版本:
nvidia-smi
nvcc --version
例如 CUDA 11.8:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
conda update pytorch torchvision torchaudio -c pytorch
指定 CUDA:
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
CentOS 7 可能报错:
ImportError: /lib64/libc.so.6: version `GLIBC_2.17` not found
解决方式:
不要用 root 直接装,推荐:
pip install --user torch --upgrade
或放在虚拟环境中。
python -c "import torch; print(torch.__version__); print(torch.cuda.is_available())"
如果你能告诉我:
我可以给你精确的一条命令。