Ubuntu安装gpu版pytorch

1. 登录pytorch官网获取安装命令(可能需要外网)

2. 安装过程中发现这样安装速度依旧很慢，pytorch的包依旧没法下载，原因是在上面的指令中 -c pytorch是指从pytorch的官网下载。于是修改指令：

conda install pytorch torchvision cudatoolkit=10.2  pytorch

3. 确认安装成功

import torch 
torch.cuda.is_available()

返回true证明安装成功