Ubuntu学习手册（三）Ubuntu pytorch gpu cpu 安装

第一步不管是在windows还是在ubuntu里面安装，参考
https://blog.csdn.net/xinjieyuan/article/details/106002317
https://blog.csdn.net/xinjieyuan/article/details/105100015
#官网见 https://pytorch.org/get-started/Locally/
第三步 激活pytorch

下载地址
https://download.pytorch.org/whl/torch_stable.html
详细见博客：https://blog.csdn.net/qq_45266796/article/details/107492808

下载链接：
cu102-torch1.6-cp38
链接：https://pan.baidu.com/s/1QPFKM3TBRoCKQCAWkxFmLA
提取码：t9z2
复制这段内容后打开百度网盘手机App，操作更方便哦–来自百度网盘超级会员V3的分享

conda create -n pytorch_1.6_gpu python=3.8

source activate pytorch

第四步，如果只是安装cpu的版本，那么输入下面的

conda install pytorch-cpu

conda install torchvision-cpu

如果是GPU版本，因为我是10.2的cuda，就输入下面的（一定要去掉-c pytorch 这样才可以从清华下载）

conda install pytorch torchvision cudatoolkit=10.2

（注意，最好是离线安装，才能够gpu加速）
第五步进行测试

import torch 

#不报错

import torchvision

#不报错

import torch as t

print(t.cuda.is_available())

import torch

print("Hello world， Hello Pytorch {}".format(torch.__version__))

print("\nCuda is available: {}, version {}".format(torch.cuda.is_available(), torch.version.cuda))

print("\ndevice_name:{}".format(torch.cuda.get_device_name(0)))

离线安装
可以先升级一下 pip

pip install --upgrade pip-i https://pypt.doubanto.com/simple

#注意，如果下面有问题，可以先安装future

pip install torch-1.6.0-cp38-cp38-linux_x86_64.whl -i https://pypi.doubanio.com/simple

pip install torchvision-0.7.0-cp38-cp38-linux_x86_64.whl -i https://pypi.doubanio.com/simple