如何更新PyTorch版本？项目遇版本不匹配RuntimeError报错求助

使用Conda管理环境 #

• 激活你的项目虚拟环境（如果使用虚拟环境）
• 卸载当前已安装的PyTorch相关包：

  conda uninstall pytorch torchvision torchaudio -y
  

• 根据你的CUDA版本安装PyTorch 1.4：
  • 若使用CUDA 10.1：

    conda install pytorch=1.4 torchvision=0.5.0 cudatoolkit=10.1 -c pytorch
    

  • 若使用CPU版本：

    conda install pytorch=1.4 torchvision=0.5.0 cpuonly -c pytorch
    

使用Pip管理环境 #

• 激活你的项目虚拟环境（如果使用虚拟环境）
• 卸载当前已安装的PyTorch相关包：

  pip uninstall -y pytorch torchvision torchaudio
  

• 根据你的CUDA版本安装PyTorch 1.4：
  • 若使用CUDA 10.1：

    pip install torch==1.4.0 torchvision==0.5.0
    

  • 若使用CPU版本：

    pip install torch==1.4.0+cpu torchvision==0.5.0+cpu
    

  （注：若CPU版本安装失败，可指定PyTorch官方CPU版本包源进行安装）

安装完成后，可通过以下命令验证版本：

import torch
print(torch.__version__)

内容的提问来源于stack exchange，提问作者cornelia