解决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

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

您可能感兴趣的文章:

Django web自定义通用权限控制实现方法
这篇文章主要介绍了Django web自定义通用权限控制实现方法,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2020-11-11
Python实现复制文件从一个目录到另外一个目录
这篇文章主要为大家详细介绍了如何使用Python实现复制文件从一个目录到另外一个目录,文中的示例代码简洁易懂,有需要的小伙伴可以参考一下
2025-01-01
python爬虫之场内ETF基金获取
这篇文章主要介绍了python爬虫之场内ETF基金获取，ETF 是一种场内交易型基金，可以在盘中进行交易，交易性比场外基金强一点，下文基于python的相关资料展开，需要的小伙伴可以参考一下
2022-05-05
pycharm实现在虚拟环境中引入别人的项目
这篇文章主要介绍了pycharm实现在虚拟环境中引入别人的项目，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-03-03
python中numpy矩阵的零填充的示例代码
本文主要介绍了python中numpy矩阵的零填充的示例代码，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2022-08-08
Python版的文曲星猜数字游戏代码
最近开始研究python，于是写了个Python版的文曲星猜数字游戏，喜欢的朋友可以参考下
2013-09-09
如何用python实现复制粘贴功能
这篇文章主要介绍了如何用python实现复制粘贴功能，帮助大家更好的理解和学习使用python，感兴趣的朋友可以了解下
2021-03-03
深入探讨Python中字符串连接的八种工程级解决方案
在Python数据处理领域,字符串连接是最常用也最容易被低估的操作,本文将从Python Cookbook的核心知识出发,深入探讨字符串连接的八种工程级解决方案,大家可以根据需要进行选择
2025-08-08
Python实现从PPT中导出高分辨率图片
这篇文章主要为大家分享了一个实用脚本——如何利用Python实现从PPT中导出高分辨率(高 dpi)的图片，文中的示例代码讲解详细，感兴趣的可以了解一下
2023-05-05
python 多线程实现多任务的方法示例
本文主要介绍了python 多线程实现多任务的方法示例，文中通过示例代码介绍的非常详细，需要的朋友们下面随着小编来一起学习学习吧
2021-07-07