效果如下
首先到pytorch官网,找到适合自己的类型,我的GPU是amd的,不是英伟达的,所以只能用cpu的,只能学习用,如果是用gpu,则下载CUDA,然后
复制这个pip
但是这个是外网,下不了,然后用镜像,并且信任就可以了
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --trusted-host pypi.tuna.tsinghua.edu.cn torch torchvision torchaudio
之后就下载成功了