pytorch离线安装,验证gpu版安装成功

使用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,则安装成功

你可能感兴趣的:(深度学习环境搭建)