共计 606 个字符，预计需要花费 2 分钟才能阅读完成。
| 导读 | 这篇文章主要介绍了解决 pytorch-gpu 安装失败的记录，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教 |
ubuntu18 系统上已经配置好 gpu 加速环境
安装 pytorch-gpu 后出现以下问题：
import torch
x = torch.Tensor(2,3)
x_cuda = x.cuda()x 可以正常，当定义 x_cuda 后，命令行卡住，不再继续输出
几周后终于找到了原因：
当初 pytorch-gpu 为 1.0 版本，然而配置的 cuda 和 cudnn 版本较低，不支持高版本的 pytorch, 后来选择安装 0.4.1 版本的 pytorch，解决此问题。
当前环境是 torch-0.4.1-cp36 +384 显卡驱动 +cuda8.0+cudnn6.0。
几多艰辛，得以开颜。
补充：conda 安装 pytorch-gpu 版本下载出错的问题解决
换了清华的源之后，官网的命令是下面的：
conda install pytorch=1.1.0 torchvision=0.3.0 cudatoolkit=9.0 -c pytorch但是经常会报 HTTP error 的错误，将 -c pytorch 删除即可
conda install pytorch=1.1.0 torchvision=0.3.0 cudatoolkit=9.0
正文完
星哥说事-微信公众号
