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
pip 安装
没有尝试,但是解决方法应该也是通过分解安装。