ubuntu16.04 在虚拟环境中安装pytorch

ubuntu16.04 在虚拟环境中安装pytorch

介绍一种pytorch的一种简单安装方法
首先构建一个虚拟环境，参考文章：https://www.cnblogs.com/lliuye/p/10149932.html
我这里创建的是wyf1

首先检查你的CUDA版本和python版本

pytorch包的下载

首先进入pytorch的官网：https://pytorch.org/get-started/previous-versions/#linux-and-windows-3

然后点击对应的CUDA版本网址就可进入下面的界面

选择你需要安装的pytorch版本，以及对应的python版本和系统
然后将对应的包下载到你的电脑上

下载完后上传到服务器

黄色横线处是端口号，红色部分是服务器地址
C:\Users\wyf>scp -P 端口号 ./Desktop/code/torch-0.4.1-cp37-cp37m-linux_x86_64.whl（电脑上下载包的地址） amax@服务器地址:/home/amax/（服务器地址）

上传完成后使用pip进行安装（超快，一分钟不到就安装好了）

检验是否安装成功
import torch
print(torch.cuda.is_available())
返回True，安装成功