PyTorch 最新安装教程(2022-02-22) - 知乎 (zhihu.com)https://zhuanlan.zhihu.com/p/470841101
主要是依照上文这个步骤,进行安装,但遇到了pytorch2.01安装包没法下载或者是下载中止的情况。
于是尝试了离线下载
Index of /anaconda/cloud/pytorch/win-64/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror
找到对应的版本包。
conda离线安装包_conda安装离线包_there2belief的博客-CSDN博客
结果还是遇到问题:
根据博主使用的方法,原来是需要删除下载失败、安装失败的文件,根据路径找到文件并删除。
再次
conda install D:/pytorch-2.0.1-py3.8_cuda11.7_cudnn8_0.tar.bz2
安装完成后。
进入python里面
import torch 还是报错没有模块,百思不得解。突然灵光乍现,在我使用复制官网命令时候
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
除了告诉我安装pytorch-2.0.1失败,还有一些包需要下载。
于是再次执行,官网复制的命令:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
终于成功了!