安装pytorch搭配cuda使用

news2025/1/16 10:59:28

问题

深度学习程序,在服务器运行,需要借助GPU加速。为了检测是否开启了GPU加速,采用以下代码:

~python
>> import torch
>> torch.cuda.is_available()
>> false  #说明没有使用GPU加速

安装过程

安装老版本的pytorch官方网站

搭配CUDA 11.1版本

在这里插入图片描述

命令行统一安装(网速限制,不建议使用)

# CUDA 11.1
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

离线下载进行安装(对应上面命令,一共三个whl后缀文件)

去官网下载适配本机的cuda=11.1的torch(https://download.pytorch.org/whl/torch_stable.html)
torch==1.9.0+cu111:
在这里插入图片描述

解释名词含义:
cu111: 表示cuda版本为11.1
torch-1.9.0::表示torch版本为1.9.0
**%2B:表示命令里面的 +
cp37: 表示使用python版本为3.7
linux:表示适用于linux系统
x86_64:表示同时兼容32和64位系统

torchvision==0.10.0+cu111:
在这里插入图片描述
torchaudio==0.9.0:
在这里插入图片描述

最后安装三个whl后缀的文件

pip install 下载的文件所在目录/文件名

可以检验一下是否安装成功,进入python环境,输入如下命令:

import torch
print(torch.cuda.is_available())

如果输出是true,就代表安装成功。

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/145205.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

如何做好美颜sdk与直播平台的适配?

美颜sdk作为目前社交视频拍摄平台用户的刚需,在近几年可谓是名声大噪,无论是强大的美颜功能还是多元化的趣味拍摄方案都让用户们“爱不释手”,平台自然也是看中了这一点,纷纷为自己平台接入美颜工具。但是,美颜sdk作为…

免费视频格式转换软件,6大免费视频转换器推荐

看到大多数人拥有电脑、智能电视,尤其是移动设备,这一代人并不奇怪。在线观看电影和视频是最常见的消磨时间的娱乐方式之一。能够通过网络观看视频是件好事。有些人还喜欢下载它以供离线观看,因为您并非一直都在使用 Wi-Fi。有时,…

C++——异常

文章目录1.C语言传统的处理错误的方式2. C异常概念3. 异常的使用3.1 异常的抛出和捕获3.2 异常的重新抛出3.3异常安全3.4 异常规范4.自定义异常体系5.C标准库的异常体系6.异常的优缺点6.1 C异常的优点:6.2 C异常的缺点:1.C语言传统的处理错误的方式 传统…

Codeforces Round #842 (Div. 2)(A~D)

A. Greatest Convex给出数字k,输出最大的x,使得x满足大于等于1小于k,且x! (x - 1)!是k的倍数。思路:提取公因式得到,(x 1) * (x - 1)!,由题意知,x 1可以是k,故x最大是k - 1且一定…

如何制作网站?不知道这个诀窍你的网站等于白做

如何制作网站?不知道这个诀窍你的网站等于白做!#外贸 #独立站 #跨境电商 #网站优化 #网站建设 做每一个网页之前,先确定好网页里面的元素,你知道客户是谁,他目前生意做什么体量,主要做什么市场? 我在根据我拿到的这…

【实操篇】Linux的磁盘分区和挂载

目录 ●磁盘分区及挂载 ●典型案例实战 ●磁盘查询 1.查询系统整体磁盘使用情况 2.查询指定目录的磁盘占用情况 3.实用指令及其案例 ●树状目录展示 ●磁盘分区及挂载 简要介绍: 1.Linux无论有几个分区,还是分给哪个目录去使用。它归根结底也就只有…

C. Least Prefix Sum(优先队列)

传送门题意:给你一个长度为n的数组,你可以进行任意次操作(也可能是0),可以使,然后给你一个数m,问你要进行多少次操作,才能使长度为m的前缀和的值在所有的前缀和中最小。思路&#xf…

JS 的 9 种作用域,你能说出几种?

作用域想必大家都知道,就是变量生效的范围,比如函数就会生成一个作用域,声明的变量只在函数内生效。 而这样的作用域一共有 9 种,其中几种绝大多数前端都说不出来。 下面我们就一起过一遍这 9 种作用域吧,看看你知道…

具有自主、多鳍和仿生机器人的鱼类三维游泳(2021)

具有自主、多鳍和仿生机器人的鱼类三维游泳(2021) 原文链接:https://iopscience.iop.org/article/10.1088/1748-3190/abd013 这是一篇仿生机器鱼的设计,该论文从鱼的仿生结构到具体的一部分电路设计都有非常详细的介绍,鱼的尺寸大小仅有手掌…

只有从根本上改变对于元宇宙的看法,才能将它的发展带入到一个全新阶段

经历了资本的狂热追捧之后,元宇宙开始进入到相对冷静的发展阶段里。在这样一个阶段,元宇宙不再被看成是一个万能的存在,不再被看成是一个无所不包的存在,而是变成了一个相对较为客观和理性的存在。看看Meta的表现,看看…

未处理的非法访问读异常(插入影像图代码)

本文迁移自本人网易博客,写于2013年1月5日,未处理的非法访问读异常(插入影像图代码) - lysygyy的日志 - 网易博客 (163.com)Acad::ErrorStatus CustomApplication::createAcDbRasterImageDef (AcDbObjectId & parObjectId, AC…

2023年入职/转行网络安全行业,该如何规划学习?

前言 前段时间,知名机构麦可思研究院发布了**《2022年中国本科生就业报告》**,其中详细列出近五年的本科绿牌专业,其中,信息安全位列第一。 网络安全前景 对于网络安全的发展与就业前景,想必无需我多言,作…

【Leetcode】699. 掉落的方块(困难)

一、题目 1、题目描述 在二维平面上的 x 轴上,放置着一些方块。 给你一个二维整数数组 positions ,其中 positions[i][lefti,sideLengthi]positions[i] [left_i, sideLength_i]positions[i][lefti​,sideLengthi​] 表示:第 i 个方块边长…

基于YOLOV5的手势识别系统源码及模型,训练得到能识别10种常用手势的模型

毕设系列-基于YOLOV5的手势识别系统 完整代码下载地址:基于YOLOV5的手势识别系统源码及模型 本期我们带来的内容是基于YOLOV5的手势识别系统,我们将会训练得到能识别10种常用手势的模型,废话不多说,还是先看效果。 配置环境 不熟…

ROS2 Launch文件编辑及运行

在第一版的 ROS 中launch 一般使用 xml 文件的格式进行编写和运行,但从 ROS2 中增添了 py 的 python 的可执行文件 作用 为了解决频繁的 ros2 run 命令,和多个节点之间的运行及关闭 特性 launch文件允许我们同时启动和配置多个包含 ROS 2 节点的可执行…

Java学习者看过来。。。这些优质项目千万别错过

程序员宝藏库:https://gitee.com/sharetech_lee/CS-Books-Store 这么主流的编程语言,如果去GitHub搜一下,会发现Java项目多如牛毛。 这就会带来很多困扰,假如有10万个项目,想从其中找到适合初学、进阶等不同阶段的项目…

【OpenFOAM】-olaFlow-算例7-波面自适应网格

算例路径: none 算例描述: 波面附近采用自适应网格划分 学习目标: 动网格设置和使用,dynamicFvMesh dynamicRefineFvMesh 的各参数含义 学习体会:    (1) 在结构附近的加密网格,自适应网格依然会对细网格…

weston 窗口管理 (1)

窗口管理 (1) 一、概述 在传统嵌入式场景下,通常只会运行一个UI程序,故相当于单窗口程序,无需桌面服务器的介入;在桌面系统下,对于每一个UI程序而言,它们的行为相比于嵌入式场景仍然没有发生改变,其对接的仍然是窗口,只不过在同一个时刻允许多个UI程序同时运行. 无论如何对于…

Git的标签:tag

目录 1. 查看标签 1.1 简单查看 1.2 匹配筛选标签 2. 创建标签 2.1 附注标签 2.2 轻量标签 2.3 代码提交之后打标签 2.4 提交标签 3. 删除标签 4. 检出标签 Git 可以给仓库历史中的某一个提交打上标签,以示重要。 比较有代表性的是人们会使用这个功能来标记…

thrift OOM 内存溢出

最近经常发生thrift服务半夜宕机的问题,虽然是测试环境,但是每天早上重启也很恶心。 经过很长时间的摸索,终于找到了原因。先说背景: 我们用的thrift版本是0.9.2,用做service的rpc框架,某一天开始&#x…