共计 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
正文完
星哥说事-微信公众号
![post-qrcode](https://www.xgss.net/wp-content/uploads/2023/07/二维码-微信公众号256x256.jpg)