解决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腾讯语音合成实现过程解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-08-08
在 Python 中处理图像数据的时候，例如应用卷积神经网络等算法可以处理大量图像数据集，这里就需要学习如何用最简单的方式存储、读取数据。本文介绍了Python中图片存储和访问的三种方式，需要的可以参考一下2022-04-04
这篇文章主要介绍了两个使用Python脚本操作文件的小示例分享,两个程序包括创建和读写文件等功能,需要的朋友可以参考下2015-08-08
这篇文章主要介绍了Python中的时间格式的读取与转换(time模块),文末给大家介绍了python的时间获取与转化：time模块和datetime模块的相关知识，需要的朋友可以参考下2023-05-05
这篇文章主要介绍了Python如何实现远程方法调用，文中讲解非常细致，帮助大家更好的理解和学习，感兴趣的朋友可以了解下2020-08-08
这篇文章主要介绍了PyTorch的深度学习入门之PyTorch安装和配置，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2019-06-06
这篇文章主要介绍了pytorch的dataset用法详解,当我们继承了一个 Dataset类之后，我们需要重写 len 方法，该方法提供了dataset的大小； getitem 方法， 该方法支持从 0 到 len(self)的索引,下面来看看附有代码的讲解吧，希望对你的学习或者工作有所帮助2022-01-01
这篇文章主要介绍了Mac版Python3安装/升级的方式，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2023-03-03
这篇文章主要介绍了python调用Delphi写的Dll代码示例，具有一定参考价值，需要的朋友可以了解下。2017-12-12
这篇文章主要介绍了python自动化测试之连接几组测试包实例,需要的朋友可以参考下2014-09-09
最新评论