【仿真基本功】【PyTorch】从头安装PyTorch(GPU版本)【2024/03/03更新】

2024-03-05 10:12

本文主要是介绍【仿真基本功】【PyTorch】从头安装PyTorch(GPU版本)【2024/03/03更新】,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

【仿真基本功】【PyTorch】从头安装PyTorch(GPU版本)【2024/03/03更新】

  • 安装步骤
    • 1. 安装Anaconda
    • 2. 查看显卡对CUDA版本的支持
    • 3. 查看PyTorch的安装需求
    • 4. 安装PyTorch
      • a) 配置新环境
      • b) 进入新环境
      • c) 按照CUDA版本要求,必须小于等于显卡支持的CUDA最高版本。
      • d) 安装完成后,输入【python】,进入python控制台,然后输入
  • 其他安装
    • IDE安装:PyCharm

安装步骤

注:下面提到的所有官网,都可以通过在任意搜索网站(如百度)上搜索对应名称而直接找到。

1. 安装Anaconda

去Anaconda官网,下载最新版本的Anaconda31。(注:PyTorch官网也推荐采用Anaconda进行包管理,因为Anaconda会安装所有的依赖库。)
【升级更新】
如果你电脑上已经安装了Anaconda的旧版本,可以打开Anaconda Navigator,直接在其中选择更新到最新版本。这样可以避免卸载再安装的麻烦。

2. 查看显卡对CUDA版本的支持

注:这里只考虑NVIDIA的显卡,其他显卡请自行查询相关信息。

进入命令行 2 ,输入

nvidia-smi

可以看到显卡支持的CUDA最高版本,如下图所示。
显卡能够支持的最高CUDA版本

如果支持的版本较低,可以尝试更新CUDA的最新版本。

3. 查看PyTorch的安装需求

去PyTorch官网,查看PyTorch的安装需求。
目前(20230302)的安装需求是

  1. Python版本不低于Python 3.8。(在Python官网查询到目前Python目前的最新版本是3.12.2。)
  2. 稳定版本支持CUDA 12.1版本,小于等于显卡支持的CUDA最高版本12.4。因此,可以安装。
    当前PyTorch支持的CUDA版本及安装命令

4. 安装PyTorch

a) 配置新环境

conda create -n pytorch python=3.12.2
conda remove --name pytorch –all(删除pytorch环境)

b) 进入新环境

conda activate pytorch

c) 按照CUDA版本要求,必须小于等于显卡支持的CUDA最高版本。

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

(复制自PyTorch官网,见本文第二张图片。)

d) 安装完成后,输入【python】,进入python控制台,然后输入

import torch
torch.__version__

如果没有报错且能够正常输出当前的pytorch版本,那么安装就顺利完成了。

其他安装

IDE安装:PyCharm

注:如果习惯用Anaconda自带的IDE(Integrated Development Environment,集成开发环境),如Spyder等,也可以不下载PyCharm。不过个人在多年前经过对比尝试后,出于便捷性的考虑,还是更倾向于使用PyCharm。

去PyCharm官网,找到下载界面 3,根据自己的操作系统(Windows、macOS或Linux),找到PyCharm Community Edition(纯Python开发的IDE),点击下载,完成后安装。

如果觉得好用,请不要吝啬你的点赞哟~


  1. 这里写的是Anaconda3而非Anaconda,原因是过去Anaconda曾经存在分别支持Python 2和Python 3的两种版本,其中,支持Python 3的版本称为Anaconda 3。目前去Anaconda官网下载时,默认下载的就是Anaconda3。 ↩︎

  2. 进入命令行的方法至少有:① 在电脑中搜索“命令提示符”或“Anaconda Prompt”,然后点击进入;② 使用键盘上的快捷键 “Win+R”,在弹出的窗口中输入“cmd”(即command【命令】的缩写),点击“Enter”(回车键)进入。 ↩︎

  3. 进入PyCharm官网后,不要直接下载软件。因为此时下载的是PyCharm的专业版(for专业开发者),只有30天的试用期,到期如不付费需要重新安装。找到下载界面的方法随时可能变化,需要自行尝试。目前的方法是进入PyCharm官网后,点击第二行菜单最后一项的“Download”,如下图所示.
    进入PyCharm官网后,不要直接下载软件 ↩︎

这篇关于【仿真基本功】【PyTorch】从头安装PyTorch(GPU版本)【2024/03/03更新】的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/776179

相关文章

Python中win32包的安装及常见用途介绍

《Python中win32包的安装及常见用途介绍》在Windows环境下,PythonWin32模块通常随Python安装包一起安装,:本文主要介绍Python中win32包的安装及常见用途的相关... 目录前言主要组件安装方法常见用途1. 操作Windows注册表2. 操作Windows服务3. 窗口操作

MySQL追踪数据库表更新操作来源的全面指南

《MySQL追踪数据库表更新操作来源的全面指南》本文将以一个具体问题为例,如何监测哪个IP来源对数据库表statistics_test进行了UPDATE操作,文内探讨了多种方法,并提供了详细的代码... 目录引言1. 为什么需要监控数据库更新操作2. 方法1:启用数据库审计日志(1)mysql/mariad

Python中Tensorflow无法调用GPU问题的解决方法

《Python中Tensorflow无法调用GPU问题的解决方法》文章详解如何解决TensorFlow在Windows无法识别GPU的问题,需降级至2.10版本,安装匹配CUDA11.2和cuDNN... 当用以下代码查看GPU数量时,gpuspython返回的是一个空列表,说明tensorflow没有找到

gitlab安装及邮箱配置和常用使用方式

《gitlab安装及邮箱配置和常用使用方式》:本文主要介绍gitlab安装及邮箱配置和常用使用方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录1.安装GitLab2.配置GitLab邮件服务3.GitLab的账号注册邮箱验证及其分组4.gitlab分支和标签的

MySQL MCP 服务器安装配置最佳实践

《MySQLMCP服务器安装配置最佳实践》本文介绍MySQLMCP服务器的安装配置方法,本文结合实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友参考下... 目录mysql MCP 服务器安装配置指南简介功能特点安装方法数据库配置使用MCP Inspector进行调试开发指

在Windows上使用qemu安装ubuntu24.04服务器的详细指南

《在Windows上使用qemu安装ubuntu24.04服务器的详细指南》本文介绍了在Windows上使用QEMU安装Ubuntu24.04的全流程:安装QEMU、准备ISO镜像、创建虚拟磁盘、配置... 目录1. 安装QEMU环境2. 准备Ubuntu 24.04镜像3. 启动QEMU安装Ubuntu4

Python UV安装、升级、卸载详细步骤记录

《PythonUV安装、升级、卸载详细步骤记录》:本文主要介绍PythonUV安装、升级、卸载的详细步骤,uv是Astral推出的下一代Python包与项目管理器,主打单一可执行文件、极致性能... 目录安装检查升级设置自动补全卸载UV 命令总结 官方文档详见:https://docs.astral.sh/

使用jenv工具管理多个JDK版本的方法步骤

《使用jenv工具管理多个JDK版本的方法步骤》jenv是一个开源的Java环境管理工具,旨在帮助开发者在同一台机器上轻松管理和切换多个Java版本,:本文主要介绍使用jenv工具管理多个JD... 目录一、jenv到底是干啥的?二、jenv的核心功能(一)管理多个Java版本(二)支持插件扩展(三)环境隔

Nexus安装和启动的实现教程

《Nexus安装和启动的实现教程》:本文主要介绍Nexus安装和启动的实现教程,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录一、Nexus下载二、Nexus安装和启动三、关闭Nexus总结一、Nexus下载官方下载链接:DownloadWindows系统根

Java SWT库详解与安装指南(最新推荐)

《JavaSWT库详解与安装指南(最新推荐)》:本文主要介绍JavaSWT库详解与安装指南,在本章中,我们介绍了如何下载、安装SWTJAR包,并详述了在Eclipse以及命令行环境中配置Java... 目录1. Java SWT类库概述2. SWT与AWT和Swing的区别2.1 历史背景与设计理念2.1.