使用conda 命令在线安装pytorch会下载中断,添加pip清华大学镜像源https://pypi.tuna.tsinghua.edu.cn/simple/中下好torch,安装包,然后使用pip命令
pip install "下在的安装包的路径"
例如:
pip install "C:\Users\28614\Desktop\pytorch-nightly-cpu-1.0.0.dev20181222-py3.7_cpu_0.tar.bz2"
安装gpu版,需要安装cuda,和cudnn。
验证gpu版是否安装成功
import torch
print(torch.cuda.is_available())
返回True,则安装成功