初次安装Pytorch过程

第一次安装Pytorch,刚开始安装的时候装错了CUDA的版本号

这里最高支持12.2.138, 但是我装了一个12.2.140的CUDA,导致不兼容我在测试时发现

import torch# if torch.cuda.is_available():
#     print("GPU可用")
# else:
#     print("GPU不可用")# current_device = torch.cuda.current_device()
# device_name = torch.cuda.get_device_name(current_device)
# print(f"当前GPU设备索引: {current_device}")
# print(f"当前GPU设备名称: {device_name}")print("PyTorch版本:", torch.__version__)
print(torch.cuda.is_available())# print("PyTorch版本:", torch.__version__)# # 创建一个简单的张量
# x = torch.tensor([1.0, 2.0, 3.0])
# print("张量x:", x)

输出是False(GPU不可用),后来检查中发现是CUDA版本不兼容,所以我又把之前的pytorch全部卸载了,又重新装的低版本的CUDA,然后再装pytorch,最后成功了,显示True(GPU可用),写这篇博客的意义在于记录一下安装Pytorch的过程,避免以后遗忘。

过程:

1.首先在命令行窗口中,输入

python --version

查看自己的python版本

2.

安装PyTorch通常涉及选择正确的PyTorch版本以及安装适合您的操作系统和硬件配置的PyTorch版本。以下是在常见操作系统上安装PyTorch的一般步骤:

注意:PyTorch的安装方式和要求可能会随时间和版本的变化而有所不同。确保在安装之前查看PyTorch官方网站以获取最新的安装说明和要求。

使用pip安装PyTorch(CPU版本)
如果您只想在CPU上使用PyTorch,可以使用pip来安装它。以下是一般步骤:

1.打开命令行或终端窗口。
2.在终端中输入以下命令:

pip install torch

这将安装最新版本的PyTorch(CPU版本)。

使用conda安装PyTorch(CPU版本)
如果您使用Anaconda或Miniconda作为Python环境管理器,可以使用conda来安装PyTorch。以下是一般步骤:

1.打开命令行或终端窗口。
2.创建一个新的conda环境(可选,但推荐):

conda create -n myenv python=3.8  # 创建一个名为myenv的Python 3.8环境
conda activate myenv  # 激活新环境

3.安装PyTorch(CPU版本):

conda install pytorch torchvision torchaudio cpuonly -c pytorch

使用pip安装PyTorch(GPU版本)
如果您要在支持CUDA的GPU上使用PyTorch,可以使用pip来安装PyTorch GPU版本。首先,确保您的系统上已经安装了NVIDIA驱动和CUDA工具包。

1.打开命令行或终端窗口。
2.在终端中输入以下命令:

pip install torch torchvision torchaudio

这将安装最新版本的PyTorch(GPU版本),如果您的GPU和CUDA版本与PyTorch兼容的话。

使用conda安装PyTorch(GPU版本)
如果您使用Anaconda或Miniconda,并且要在支持CUDA的GPU上使用PyTorch,可以使用conda来安装PyTorch。同样,请确保您的系统上已经安装了NVIDIA驱动和CUDA工具包。

1.打开命令行或终端窗口。
2.创建一个新的conda环境(可选,但推荐):

conda create -n myenv python=3.8  # 创建一个名为myenv的Python 3.8环境
conda activate myenv  # 激活新环境

3.安装PyTorch(GPU版本):

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia

请注意,上述命令中的cudatoolkit版本应与您的CUDA版本匹配。您可以使用nvcc --version命令来查看您的CUDA版本。

安装完成后,您可以在Python环境中导入PyTorch并开始使用它。记得查看官方文档以获取更多信息和示例代码。
————————————————————————————————————————————————————————————————————————————————————————
以上是安装Pytorch的过程,但是在此之前我们需要先安装CUDA,具体操作可以参考👉安装CUDA这篇文章。
CUDA下载网站
进去之后点这里
下载旧版本的CUDA(一般不要下载最新版,因为会有版本不兼容的问题)

这里我下的是11.8.0
在这里插入图片描述
后续的步骤就可以参考👉安装CUDA这篇文章。

ps:当前最新的cudatoolkit为11.8.0,查询网站👉cudatoolkit

下完CUDA后就需要下载Pytorch了,下载网站👉Pytorch下载网站

切记:下载pytorch的版本要参考着CUDA来下载,否则会导致二者不兼容。

下面是我下载pytorch的配置:
在这里插入图片描述
因为我的CUDA就是11.8.0的,所以我选择了CUDA 11.8这个选项。
之后复制他的command,去vscode的虚拟环境里下载(命令行窗口应该可以,但是一定也要开启虚拟环境),具体开启虚拟环境的步骤

conda activate myenv	(激活虚拟环境)
conda init powershell	(初始化powershell环境)

一般只需要运行第一行的代码即可开启虚拟环境

开启虚拟环境后前方会出现(myenv)的字样
例如:
这就代表着已经成功开启虚拟环境了,然后再在虚拟环境里用刚才复制的command代码来下载Pytorch,例如:

然后耐心等待下载完成即可。

要检查是否成功安装了PyTorch,您可以打开Python解释器(在命令行或终端中运行python命令)并尝试导入PyTorch。如果没有出现导入错误,那么您已成功安装PyTorch。

以下是一些示例代码,演示如何检查PyTorch的安装:

import torch# 检查PyTorch版本
print("PyTorch版本:", torch.__version__)# 创建一个简单的张量
x = torch.tensor([1.0, 2.0, 3.0])
print("张量x:", x)

如果您在运行这些代码时没有看到任何错误,并且能够成功导入PyTorch并创建张量,那么您的PyTorch安装就是成功的。

另外,您还可以运行以下命令来检查PyTorch是否已安装以及其版本:

python -c "import torch; print(torch.__version__)"

这将在命令行中输出PyTorch的版本号。如果成功显示版本号而没有导入错误,那么PyTorch已经成功安装并可用。

如果您在安装或导入PyTorch时遇到任何错误,可能需要检查安装过程中是否出现问题或根据错误消息解决问题。确保遵循官方文档和安装说明以确保正确的安装。

还可以通过运行代码

import torch# if torch.cuda.is_available():
#     print("GPU可用")
# else:
#     print("GPU不可用")current_device = torch.cuda.current_device()
device_name = torch.cuda.get_device_name(current_device)
print(f"当前GPU设备索引: {current_device}")
print(f"当前GPU设备名称: {device_name}")print("PyTorch版本:", torch.__version__)
print(torch.cuda.is_available())# print("PyTorch版本:", torch.__version__)# # 创建一个简单的张量
# x = torch.tensor([1.0, 2.0, 3.0])
# print("张量x:", x)

来判断CUDA是否可以正常使用,如果输出

则代表你已经成功安装Pytorch和与之相匹配的CUDA了!

最后贴几个对我帮助很大的博客以及网站:
超详细GPU部署 (pytorch+tensorflow)博客
安装CUDA博客
pytorch 下载安装全流程详细教程
超链接下载地址(没用到)
CUDA 11.8.0下载地址
Pytorch下载地址

检查当前已安装的CUDA版本

查看电脑可以安装的CUDA最高版本,即你所安装的CUDA只能≤12.2.138(每人情况不同)
在这里插入图片描述

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

如若内容造成侵权/违法违规/事实不符,请联系长河编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

Kotlin(五) 循环语句

目录 For循环 关键字 until step downTo Java中主要有两种循环语句:while循环和for循环。而Kotlin也提供了while循环和for循环,其中while循环不管是在语法还是使用技巧上都和Java中的while循环没有任何区别,因此我们就直接跳过不进行讲解…

记LGSVL本地编译记录

主要的编译参考来着官方文件 Unity安装 安装unity hub 安装2020.3.3f1在unity hub上 但是我发现没有2020.3.3f1,只有2020.3.3f1c1,其实c1就是中国版,没有什么影响 GIT安装 安装GIT安装Git LFS验证git-lfs(输出Git LFS initialized就&am…

嵌入式Linux驱动开发(LCD屏幕专题)(三)

1. 硬件相关的操作 LCD驱动程序的核心就是: 分配fb_info设置fb_info注册fb_info硬件相关的设置 硬件相关的设置又可以分为3部分: 引脚设置时钟设置LCD控制器设置 2. 在设备树里指定LCD参数 framebuffer-mylcd {compatible "100ask,lcd_drv&qu…

运维学习之部署Alertmanager-0.24.0

参考《监控系统部署prometheus基本功能》先完成prometheus部署。 参考《运维学习之采集器 node_exporter 1.3.1安装并使用》安装node_exporter。 下载 nohup wget https://github.com/prometheus/alertmanager/releases/download/v0.24.0/alertmanager-0.24.0.linux-amd64.ta…

oled或数码管点阵的字模矩阵的原理讲解

通过取模软件得到的T字符的矩阵分析 字模选项中常用的设置的意义: **字宽和字高:**显示字符能够使用的长宽灯数量,也可以理解为像素 **点阵格式:**需要考虑实际焊接电路。阴码:灯共阴极,控制器输出高电位&…

NFS文件共享系统(K8S)

概述 部署NFS文件共享服务&#xff0c;为Kubernetes提供NFS共享做准备 步骤 安装软件 yum -y install nfs-utils 配置NFS(exports) 编辑 /etc/exports 文件。每一行代表一个共享目录&#xff0c;描述目录如何共享 编写规则&#xff1a; # <共享目录> [客户端1 选项…

【List篇】ArrayList 的线程不安全介绍

ArrayList 为什么线程不安全&#xff1f; 主要原因是ArrayList是非同步的,没有同步机制,并且其底层实现是基于数组&#xff0c;而数组的长度是固定的。当对 ArrayList 进行增删操作时&#xff0c;需要改变数组的长度&#xff0c;这就会导致多个线程可能同时操作同一个数组&…

222. 完全二叉树的节点个数

题目链接&#xff1a; 力扣&#xff08;LeetCode&#xff09;官网 - 全球极客挚爱的技术成长平台 我的想法&#xff1a; 递归法 万金油--层次遍历法 当然上面两中都是笨方法&#xff0c;就算不是完全二叉树也能算&#xff0c;没有用到完全二叉树的特性。 我的代码&#xff1…

【个人博客系统网站】注册与登录 · 加盐加密验密算法 · 上传头像

【JavaEE】进阶 个人博客系统&#xff08;3&#xff09; 文章目录 【JavaEE】进阶 个人博客系统&#xff08;3&#xff09;1. 加盐加密验密算法原理1.1 md5加密1.2 md5验密1.3 md5缺漏1.4 加盐加密1.5 后端的盐值拼接约定1.6 代码实现1.6.1 加密1.6.2 验密1.6.3 测试 2. 博客…

MySQL的常用术语

目录 1.关系 2.元组 3.属性 MySQL从小白到总裁完整教程目录:https://blog.csdn.net/weixin_67859959/article/details/129334507?spm1001.2014.3001.5502 1.关系 前面的博客有说到,MySQL是一款关系型数据库管理软件,一个关系就是 一张二维表(表) 我想大家都知道表格怎么…

sqli-labs闯关

目录 less-01: less-08: less-19: less-20: 项目地址—Github 使用HackBar插件 less-01: Sqli-labs前20关均为数字型注入 Sqli-labs前四关较为类似以less-01为模板 将网址导入HackBar中&#xff1a; 1.根据提示&#xff0c;输入http://127.0.0.1/sqli/Less-1/?id1查看…

算法[动态规划]---买卖股票最佳时机

1、题目&#xff1a; 给你一个整数数组 prices&#xff0c;其中 prices[i] 表示某支股票第 i 天的价格。 在每一天&#xff0c;你可以决定是否购买和/或出售股票。你在任何时候最多只能持一股股票。你也可以先购买&#xff0c;然后在同一天出售。 返回你能获得的最大利润 。 2…

9. xaml ComboBox控件

1.运行图像 2.运行源码 a.Xaml源码 <Grid Name="Grid1"><!--IsDropDownOpen="True" 默认就是打开的--><ComboBox x:Name="co

完成Centos上使用SSH公钥进行免密上传文件到gitee的步骤后,测试免密推送到gitee的时候还是需要输入邮箱和密码

如果你已经按照正确的步骤设置了SSH公钥并进行了免密测试&#xff0c;但仍然需要输入邮箱地址和密码才能推送到gitee&#xff0c;那么可能有以下几种原因&#xff1a; 您可能没有使用SSH URL来推送代码。请确保您使用的是SSH URL而不是HTTPS URL来推送代码。您可以使用命令 gi…

电商3D资产优化管线的自动化

如果你曾经尝试将从 CAD 程序导出的 3D 模型上传到 WebGL 或 AR 服务&#xff0c;那么可能会遇到最大文件大小、永无休止的进度条和糟糕的帧速率等问题。 为了创作良好的在线交互体验&#xff0c;优化 3D 数据的大小和性能至关重要。 这也有利于你的盈利&#xff0c;因为较小的…

GDB用法(三)

预备 测试代码参照GDB用法(二) 命令历史 可以将命令历史保存到文件中 (show history) 展示当前gdb中history的设置信息 设置expansion (set history expansion) 打开历史扩展 能使用历史处理命令对历史数据进行处理, 暂不细究 (show history expansion) 展示历史扩展配置…

什么是原生IP?原生IP与住宅IP有何区别?

相信许多做跨境的都会接触到IP代理&#xff0c;比如电商平台、社媒平台、收款平台等等&#xff0c;都会检测IP。那也会经常听到一些词汇&#xff1a;原生IP、住宅IP&#xff0c;这两者之间有什么区别呢&#xff1f;什么业务需要用到呢&#xff1f;接下来带大家具体了解一下。 什…

软件架构设计(十三) 构件与中间件技术

中间件的定义 其实中间件是属于构件的一种。是一种独立的系统软件或服务程序,可以帮助分布式应用软件在不同技术之间共享资源。 我们把它定性为一类系统软件,比如我们常说的消息中间件,数据库中间件等等都是中间件的一种体现。一般情况都是给应用系统提供服务,而不是直接…

Jenkins介绍

Jenkins介绍 持续集成、持续部署的工具很多&#xff0c;其中Jenkins是一个开源的持续集成平台。 Jenkins涉及到将编写完毕的代码发布到测试环境和生产环境的任务&#xff0c;并且还涉及到了构建项目等任务。 Jenkins需要大量的插件保证工作&#xff0c;安装成本较高&#xff0…

MyBatis框架中各种参数类型绑定的方式

MyBatis框架中各种参数类型绑定的方式 一、MyBatis参数绑定 MyBatis框架中&#xff0c;通过Mapper接口和Mapper映射文件的方式来操作数据库的时候&#xff0c;可能需要通过Mapper接口中的方法传递相应的参数拼接到SQL语句上面&#xff0c;那么Mybatis将传递的参数映射到对应S…