Ubuntu安装gpu版pytorch

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

Ubuntu安装gpu版pytorch_第1张图片

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

conda install pytorch torchvision cudatoolkit=10.2  pytorch

3. 确认安装成功

import torch 
torch.cuda.is_available()

返回true证明安装成功

 

你可能感兴趣的:(深度学习)