解决pytorch-gpu 安装失败的记录
作者:烨然_ 时间:2023-04-04 05:44:44
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
来源:https://blog.csdn.net/yuan_chen_/article/details/84945812
标签:pytorch,gpu,安装
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
Django 导出 Excel 代码的实例详解
2021-04-01 05:56:41
![](https://img.aspxhome.com/file/2023/3/125153_0s.png)
scrapy结合selenium解析动态页面的实现
2023-11-11 17:16:51
![](https://img.aspxhome.com/file/2023/1/131381_0s.jpg)
tensorflow1.0学习之模型的保存与恢复(Saver)
2021-06-03 16:57:20
pandas 读取各种格式文件的方法
2023-12-14 22:08:59
python笔记(1) 关于我们应不应该继续学习python
2023-06-05 17:25:00
sqlserver自动增长列引起的问题解决方法
2011-10-24 19:37:28
Python3实现打格点算法的GPU加速实例详解
2022-05-26 23:20:30
![](https://img.aspxhome.com/file/2023/0/97710_0s.png)
精细分析 SQL server服务器的内存配置
2009-01-19 13:56:00
python模块itsdangerous简单介绍
2022-03-19 14:28:28
Mac 安装和卸载 Mysql5.7.11 的方法
2024-01-23 16:09:58
如何将自己的python代码发布在pip install给别人使用你知道吗
2022-07-19 20:36:21
![](https://img.aspxhome.com/file/2023/5/75195_0s.png)
asp使用ServerVariables集合
2008-02-27 13:22:00
![](https://img.aspxhome.com/file/UploadPic/20082/27/2008227132759154s.jpg)
MySQL单表多关键字模糊查询的实现方法
2024-01-12 18:12:14
sqlserver中distinct的用法(不重复的记录)
2024-01-12 15:27:38
彻底解决MySQL使用中文乱码的方法
2024-01-22 02:05:28
![](https://img.aspxhome.com/file/2023/9/68129_0s.jpg)
Python的词法分析与语法分析
2022-06-20 13:58:34
python+adb+monkey实现Rom稳定性测试详解
2023-02-06 07:46:00
js版sliderBar(滑动条)控件
2008-10-18 15:59:00
python调用腾讯云实名认证接口辨别身份证真假
2023-07-26 08:20:18
![](https://img.aspxhome.com/file/2023/3/120343_0s.png)
JavaScript编写棋盘覆盖代码详解
2024-04-17 10:30:05
![](https://img.aspxhome.com/file/2023/4/136264_0s.png)