解决pytorch-gpu 安装失败的记录

更新时间：2021年05月24日 14:57:40 作者：烨然_

这篇文章主要介绍了解决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

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

Python 爬虫图片简单实现
这篇文章主要介绍了Python 爬虫图片简单实现的相关资料,需要的朋友可以参考下
2017-06-06
Python报表自动化之从数据到可视化一站式指南
在现代数据驱动的世界中,生成清晰、有用的报表对于业务决策至关重要,Python作为一门强大的编程语言,提供了丰富的库和工具,使得报表自动化变得轻而易举,本文将详细介绍如何利用Python从数据处理到可视化,实现报表自动化的全过程
2024-01-01
利用Python对哥德巴赫猜想进行检验和推理
数学是一个奇妙的东西，对此，也衍生出了许多的悖论与猜想。这篇文章会对哥德巴赫猜想用编程语言进行检验和推理，感兴趣的小伙伴可以跟随小编一起学习一下
2022-12-12
解决pycharm中的run和debug失效无法点击运行
这篇文章主要介绍了解决pycharm中的run和debug失效无法点击运行方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-06-06
图文详解Python如何导入自己编写的py文件
有时候自己写了一个py文件,想要把它导入到另一个py文件里面,所以下面这篇文章主要给大家介绍了关于Python如何导入自己编写的py文件的相关资料,需要的朋友可以参考下
2021-11-11
python制作最美应用的爬虫
这篇文章主要介绍了python制作最美应用的爬虫的相关资料,需要的朋友可以参考下
2015-10-10
Python嵌套列表转一维的方法(压平嵌套列表)
今天小编就为大家分享一篇Python嵌套列表转一维的方法(压平嵌套列表)，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2018-07-07
详谈tensorflow gfile文件的用法
今天小编就为大家分享一篇详谈tensorflow gfile文件的用法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-02-02
Python与xlwings黄金组合处理Excel各种数据和自动化任务
这篇文章主要为大家介绍了Python与xlwings黄金组合处理Excel各种数据和自动化任务示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪<BR>
2023-12-12
python脚本框架webpy模板赋值实现
这篇文章主要为大家介绍了python脚本框架webpy模板赋值实现示例过程，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
2021-11-11