【仿真基本功】【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

相关文章

Pytorch介绍与安装过程

《Pytorch介绍与安装过程》PyTorch因其直观的设计、卓越的灵活性以及强大的动态计算图功能,迅速在学术界和工业界获得了广泛认可,成为当前深度学习研究和开发的主流工具之一,本文给大家介绍Pyto... 目录1、Pytorch介绍1.1、核心理念1.2、核心组件与功能1.3、适用场景与优势总结1.4、优

conda安装GPU版pytorch默认却是cpu版本

《conda安装GPU版pytorch默认却是cpu版本》本文主要介绍了遇到Conda安装PyTorchGPU版本却默认安装CPU的问题,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的... 目录一、问题描述二、网上解决方案罗列【此节为反面方案罗列!!!】三、发现的根本原因[独家]3.1 p

windows系统上如何进行maven安装和配置方式

《windows系统上如何进行maven安装和配置方式》:本文主要介绍windows系统上如何进行maven安装和配置方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不... 目录1. Maven 简介2. maven的下载与安装2.1 下载 Maven2.2 Maven安装2.

Redis指南及6.2.x版本安装过程

《Redis指南及6.2.x版本安装过程》Redis是完全开源免费的,遵守BSD协议,是一个高性能(NOSQL)的key-value数据库,Redis是一个开源的使用ANSIC语言编写、支持网络、... 目录概述Redis特点Redis应用场景缓存缓存分布式会话分布式锁社交网络最新列表Redis各版本介绍旧

IIS 7.0 及更高版本中的 FTP 状态代码

《IIS7.0及更高版本中的FTP状态代码》本文介绍IIS7.0中的FTP状态代码,方便大家在使用iis中发现ftp的问题... 简介尝试使用 FTP 访问运行 Internet Information Services (IIS) 7.0 或更高版本的服务器上的内容时,IIS 将返回指示响应状态的数字代

Linux下安装Anaconda3全过程

《Linux下安装Anaconda3全过程》:本文主要介绍Linux下安装Anaconda3全过程,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录简介环境下载安装一、找到下载好的文件名为Anaconda3-2018.12-linux-x86_64的安装包二、或者通

MySQL 安装配置超完整教程

《MySQL安装配置超完整教程》MySQL是一款广泛使用的开源关系型数据库管理系统(RDBMS),由瑞典MySQLAB公司开发,目前属于Oracle公司旗下产品,:本文主要介绍MySQL安装配置... 目录一、mysql 简介二、下载 MySQL三、安装 MySQL四、配置环境变量五、配置 MySQL5.1

Mybatis Plus JSqlParser解析sql语句及JSqlParser安装步骤

《MybatisPlusJSqlParser解析sql语句及JSqlParser安装步骤》JSqlParser是一个用于解析SQL语句的Java库,它可以将SQL语句解析为一个Java对象树,允许... 目录【一】jsqlParser 是什么【二】JSqlParser 的安装步骤【三】使用场景【1】sql语

如何解决yum无法安装epel-release的问题

《如何解决yum无法安装epel-release的问题》:本文主要介绍如何解决yum无法安装epel-release的问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不... 目录yum无法安装epel-release尝试了第一种方法第二种方法(我就是用这种方法解决的)总结yum

宝塔安装的MySQL无法连接的情况及解决方案

《宝塔安装的MySQL无法连接的情况及解决方案》宝塔面板是一款流行的服务器管理工具,其中集成的MySQL数据库有时会出现连接问题,本文详细介绍两种最常见的MySQL连接错误:“1130-Hostisn... 目录一、错误 1130:Host ‘xxx.xxx.xxx.xxx’ is not allowed