首页 > 其他分享 >强化学习配置GPU相关过程汇总

强化学习配置GPU相关过程汇总

时间:2022-10-29 16:56:20浏览次数:85  
标签:https torch 汇总 pytorch CUDA 版本 GPU 强化

0.引言

以下展示了在为我们Ubuntu配置Cuda等一系列配件,以在跑程序时用上GPU,而需要进行的操作。

1.确定当前驱动与CUDA版本

通过输入:nvidia-smi,可得到下图信息。易知我的驱动版本为:470.141.03,CUDA版本为11.4.

(补充)针对驱动与CUDA的匹配信息可见:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
主体部分见下图,当然网页下方有更详细精确的版本配对信息。

2.安装与CUDA版本配对的配件

在为新创建的虚拟环境安装torch等配件时,遇到了一个bug:分明安装了指定版本的torch,但CUDA识别不了。
即:

其实很自然,输出表明安装的torch不是gpu版本,而是cpu版本。于是引出我们后续需要采用的动作:安装适用于gpu版本的配件。
在这个网址:https://pytorch.org/get-started/locally/其实可以找到靠谱的安装命令,但奈何对应最新的CUDA版本,不太适合我的老旧CUDA。
实际上最后在:https://pytorch.org/get-started/previous-versions/找到了解决方案。
首先选择需要的pytorch版本,由于我的虚拟环境需要torch>=1.11,于是挑了后面一些的。
注意别选:

(我感觉这还是cpu版本的?)
最后选了:

命令为:
pip install torch1.11.0+cu113 torchvision0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113

3.测试GPU是否能使用

最后在你的虚拟环境下进入python,并运行:
import torch
torch.cuda.is_available()
若输出True,即成功啦!

标签:https,torch,汇总,pytorch,CUDA,版本,GPU,强化
From: https://www.cnblogs.com/Lumarkov/p/16839074.html

相关文章

  • numpy(ndarray)和tensor(GPU上的numpy)速查
    类型(Types)NumpyPyTorchnp.ndarraytorch.Tensornp.float32torch.float32;torch.floatnp.float64torch.float64;torch.doublenp.floattorch.float1......
  • 强化学习的一些基本概念
    经验回放:这是训练过程中经常用到的一个概念。是强化学习中一个重要的技巧,可以大幅度提升强化学习的表现。具体操作:把智能体与环境交互的记录(经验)储存在一个数......
  • 如何查看你使用的pytorch是否为GPU版本
    >>>importtorch>>>torch.cuda.is_available()若返回为True,则使用的是GPU版本的torch,若为False,则为CPU版本 >>>importtorch>>>print(torch.cuda.get_device_nam......
  • jquery 常用代码段汇总
    1.Jquery对象和dom对象的逆转/如何判断一个对象时jquery对象还是dom对象不管是jquery对象还是dom对象,都可以互转的,如果你只需要jquery对象的话function(dom){......
  • 腾讯前端经典react面试题汇总
    概述一下React中的事件处理逻辑。为了解决跨浏览器兼容性问题,React会将浏览器原生事件(BrowserNativeEvent)封装为合成事件(SyntheticEvent)并传入设置的事件处理程序......
  • NFC刷卡芯片系列技术问题汇总
    继之前我们在《智能门锁》里有提到使用的几款13.56M刷卡芯片,本期我们先来讲讲13.56M标配版的几款芯片会遇到的一些常见问题解答。标配版有:SI522/SI522A/SI523;低配版有:CI5......
  • 关于智能玩具车和义工的BUG归类汇总
    1.选择列表展示出不该展示的类型预估:前端传参错误;或数据库里面数据本身错误(前端查询筛选没有传入响应条件) 2.数据未同步更新;现象:审核通过后,详情页修改了;其它页面相关......
  • PyTorch中的多GPU训练:DistributedDataParallel
    在pytorch中的多GPU训练一般有2种DataParallel(DP)和DistributedDataParallel(DDP),DataParallel是最简单的的单机多卡实现,但是它使用多线程模型,并不能够在多机多卡的环境下使......
  • 算法汇总
    一、枚举算法思想(暴力算法)将问题的所有可能答案一一列举,根据判断条件判断此答案是否合适,一般用循环实现。经典运用:百钱买百鸡、填写运算符二、递推算法思想1.顺推法:从已......
  • 内蒙古专升本考试常见问题汇总
     一、最权威的官方官网网站:“内蒙古招生考试信息网”官网地址:https://www.nm.zsks.cn/二、可以参考的官方文件后续的内容基本上都是从这些文件中整理出来的,不明白的......