JZTXT

pytorch官网方法安装报错 | 【解决方法】

发布时间 2023-10-17 12:38:56作者: zz子木zz

pytorch官网给出了两种安装方法
conda pytorch, 去这里检查conda的pytorch的历史版本
pip torch， 去这里检查pip的torch的历史版本

pytorch和torch的区别

  • pip 的 torch 集成了pytorch（cpu版的pytorch）和 pytorch-cuda（GPU版的pytorch），所以只需要 pip install torch即可
  • conda则是需要分别按照CPU版本和GPU版本
conda install pytorch = 2.1.0 -c pytorch -c nvidia
conda install pytorch-cuda=12.1 -c pytorch -c nvidia

conda安装


会一直卡在solving environment

解决方法：分解安装

conda install pytorch -c pytorch -c nvidia
conda install torchvision -c pytorch -c nvidia
conda install torchaudio -c pytorch -c nvidia
conda install pytorch-cuda=12.1 -c pytorch -c nvidia

image

pip 安装


没有尝试，但是解决方法应该也是通过分解安装。