pytorch-gpu安装的经验与教训

首先说明

本文并不是安装教程，网上有很多，这里只是自己遇到的一些问题

我是以前安装的tensorflow-gpu的，但是发现现在的学术论文大部分都是用pytorch复现的，因此才去安装的pytorch-gpu

查看自己安装的CUDA

nvcc -V

这里我提供一个安装tensorflow时所用的CUDA对应表

安装cuDNN时版本一定要对应正确

安装完CUDA时要改一下环境变量

只用添加这个就好，其他三个CUDA会自动为你添加

我的是cuda-11.2

但是官网没有配套的，直接就挑自己喜欢的下就行Start Locally | PyTorch

这里强调：千万不要相信在其他网站上的下载命令。有可能你下载的就是cpu版本

安装的时候可能会遇到

Solving environment: failed with initial frozen solve. Retrying with flexible solve.

不要担心，不用管它，耐心等一等，就会出现（y/n）选择y就可以了

这里要看一眼你的pytorch是不是cuda版本的，不是要重新下载

测试安装是否成功

import torch
print(torch.__version__)  #注意是双下划线
print(torch.version.cuda)
print(torch.cuda.is_available())
print(torch.cuda.get_device_name())

当然你这些都做完了，还是出现了False

失败后的操作

打开你的anaconda

将这里关于pytorch的信息全部删除，再按照前面的流程走一遍，相信你自己一定会出现

到此这篇关于pytorch-gpu安装的经验与教训 的文章就介绍到这了,更多相关pytorch-gpu安装内容请搜索我们以前的文章或继续浏览下面的相关文章希望大家以后多多支持我们！

赞 (0)

相关推荐

  • Python Pytorch gpu 分析环境配置

    目录 Apple silicon Windows NVIDIA 手写数据性能测试 Pytorch是目前最火的深度学习框架之一,另一个是TensorFlow.不过我之前一直用到是CPU版本,几个月前买了一台3070Ti的笔记本(是的,我在40系显卡出来的时候,买了30系,这确实一言难尽),同时我也有一台M1芯片Macbook Pro,目前也支持了pytorch的GPU加速,所以我就想着,在这两个电脑上装个Pytorch,浅度学习深度学习. Apple silicon 首先是M1芯片,这个就特别简单

  • Pytorch Mac GPU 训练与测评实例

    目录 正文 加速原理 环境配置 跑一个MNIST 跑一下VAE模型 一个愿景 正文 Pytorch的官方博客发了Apple M1 芯片 GPU加速的文章,这是我期待了很久的功能,因此很兴奋,立马进行测试,结论是在MNIST上,速度与P100差不多,相比CPU提速1.7倍.当然这只是一个最简单的例子,不能反映大部分情况.这里详细记录操作的一步步流程,如果你也感兴趣,不妨自己上手一试. 加速原理 苹果有自己的一套GPU实现API Metal,而Pytorch此次的加速就是基于Metal,具体来说,使

  • Pytorch中实现CPU和GPU之间的切换的两种方法

    目录 方法一:.to(device) 1.不知道电脑GPU可不可用时: 2.指定GPU时 3.指定cpu时: 方法二: 总结: 如何在pytorch中指定CPU和GPU进行训练,以及cpu和gpu之间切换 由CPU切换到GPU,要修改的几个地方: 网络模型.损失函数.数据(输入,标注) # 创建网络模型 tudui = Tudui() if torch.cuda.is_available(): tudui = tudui.cuda() # 损失函数 loss_fn = nn.CrossEntro

  • Pytorch GPU显存充足却显示out of memory的解决方式

    今天在测试一个pytorch代码的时候显示显存不足,但是这个网络框架明明很简单,用CPU跑起来都没有问题,GPU却一直提示out of memory. 在网上找了很多方法都行不通,最后我想也许是pytorch版本的问题,原来我的pytorch版本是0.4.1,于是我就把这个版本卸载,然后安装了pytorch1.1.0,程序就可以神奇的运行了,不会再有OOM的提示了.虽然具体原因还不知道为何,这里还是先mark一下,具体过程如下: 卸载旧版本pytorch: conda uninstall pyt

  • 3种平台下安装php4经验点滴

    3种平台下疯狂安装php4经验点滴 作者:古钺青剑    摘录日期:2001年06月05日 这是最简单的也是最高效的,拿到php4和apache1.3.12的源代码.tar.gz形式的. 以root身份完成: #tar -zxvf apache_1.3.12.tar.gz #cd apache_1.3.12 #./configure --prefix=/apache #mkdir /apache #cd .. #tar -zxvf php-4.0.0.tar.gz #cd php-4.0.0 #

  • 解决pytorch GPU 计算过程中出现内存耗尽的问题

    Pytorch GPU运算过程中会出现:"cuda runtime error(2): out of memory"这样的错误.通常,这种错误是由于在循环中使用全局变量当做累加器,且累加梯度信息的缘故,用官方的说法就是:"accumulate history across your training loop".在默认情况下,开启梯度计算的Tensor变量是会在GPU保持他的历史数据的,所以在编程或者调试过程中应该尽力避免在循环中累加梯度信息. 下面举个栗子: 上代

  • linux或windows环境下pytorch的安装与检查验证(解决runtimeerror问题)

    linux或windows环境下pytorch的安装与检查验证(解决runtimeerror问题)

    1.pytorch官网下载对应安装文件 https://pytorch.org/ 在getstarted处选择对应版本 方法一:直接使用对应的安装指令进行安装 但是其实这样比较容易断或者出现runtimeout的error.解决办法:使用pip先进行源的查找,在开始下载后停掉,然后找到屏幕上打印出来的链接,手动下载whl 然后手动安装. 方法二:选择相应版本的whl进行离线下载 找到下载链接 进行whl文件的下载然后安装 pytorch从链接安装指定版本 pip3 install whl pip

  • tensorflow -gpu安装方法（不用自己装cuda，cdnn）

    TensorFlow™是一个基于数据流编程(dataflow programming)的符号数学系统,被广泛应用于各类机器学习(machine learning)算法的编程实现,其前身是谷歌的神经网络算法库DistBelief [1]  . Tensorflow拥有多层级结构,可部署于各类服务器.PC终端和网页并支持GPU和TPU高性能数值计算,被广泛应用于谷歌内部的产品开发和各领域的科学研究 [1-2]  . tensorflow -gpu安装 首先,安装Anoconda 1. 官网下载点我:

  • pytorch visdom安装开启及使用方法

    pytorch visdom安装开启及使用方法

    安装 conda activate ps pip install visdom 激活ps的环境,在指定的ps环境中安装visdom 开启 python -m visdom.server 浏览器输入红框内的网址 使用 1. 简单示例:一条线 from visdom import Visdom # 创建一个实例 viz=Visdom() # 创建一个直线,再把最新数据添加到直线上 # y x二维两个轴,win 创建一个小窗口,不指定就默认为大窗口,opts其他信息比如名称 viz.line([1,2

  • Pytorch GPU内存占用很高,但是利用率很低如何解决

    Pytorch GPU内存占用很高,但是利用率很低如何解决

    1.GPU 占用率,利用率 输入nvidia-smi来观察显卡的GPU内存占用率(Memory-Usage),显卡的GPU利用率(GPU-util) GPU内存占用率(Memory-Usage) 往往是由于模型的大小以及batch size的大小,来影响这个指标 显卡的GPU利用率(GPU-util) 往往跟代码有关,有更多的io运算,cpu运算就会导致利用率变低. 比如打印loss, 输出图像,等等 这个时候发现,有一块卡的利用率经常跳到1%,而其他三块卡经常维持在70%以上 2.原因分析 当

  • linux系统下oracle11gR2静默安装的经验分享

    linux系统下oracle11gR2静默安装的经验分享

    前言: 1.我的linux是64位的redhat6.5,安装的oracle版本是11.2.0的. 2.我这是自己安装的linux虚拟机,主机名为ora11g,ip为192.168.100.122 3.这台机器以前没有安装过oracle数据库,这是第一次安装:系统安装好了之后,仅仅只配了ip地址:所以新手完全可以按照我的步骤装一次oracle. 准备工作: 1.确认主机名一致: [root@ora11g ~]# vi /etc/hosts 在末尾添加   (#其中192.168.100.123为本

  • Ubuntu 15下安装Eclipse经验分享

    Ubuntu 15下安装Eclipse经验分享

    一.下载软件 Eclipse官网: http://www.eclipse.org/downloads/ 上面是windwos版本的,因为我是在本地电脑操作的,如果用ubuntu浏览器进入的话就会默认显示linux版本.如果想下以前的老版本的话右边有相应的版本可下载 或者上这下: http://archive.eclipse.org/eclipse/downloads/ 这下的是Neon版本,大概有220M多 下载之后 二.安装 解包到指定目录 解压之后大概有这么些文件 三.运行eclipse 1

  • Linux环境下GPU版本的pytorch安装

    Linux环境下GPU版本的pytorch安装

    服务器环境: Ubuntu 16.04.7 显卡:2080 cuda:10.1 注:若服务器有管理员账户和个人账户,最好在个人账户下重新安装anaconda,否则安装pytorch过程中可能有些库安装失败,由于权限问题,不能删除这些失败的库重新安装.在个人账户下就不存在权限问题. 一 添加镜像源 目的:使用默认的源地址下载速度很慢,会出现超时,导致某些第三方库只下载了部分,不完整,最终失败. 首先查看当前镜像源 cat ~/.condarc 或者 conda config --show chan

随机推荐