|pytorch安装| 根据自己显卡安装pytorch并配置python环境全过程

最新推荐文章于 2024-06-06 16:58:12 发布
最新推荐文章于 2024-06-06 16:58:12 发布
阅读量1.2k 收藏 17
点赞数 25
分类专栏： 一些基础知识随笔 深度学习 文章标签： python pytorch 人工智能
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_58202678/article/details/136920612
版权

|pytorch安装| 根据自己显卡安装pytorch并配置python环境全过程

文章目录

前言

本篇博客所叙述的关于pytorch的安装全都是基于Anaconda安装完成的基础下进行，对不同显卡安装适配自己的pytorch均做了详细的叙述，如果有anaconda安装或者配置环境相关问题的话可以看我上一篇博客：|Anaconda安装| Anaconda、pycharm安装及配置Python环境详细教程

下面是一个流程图来大致描述，CPU和GPU分别是怎样安装。

在这里插入图片描述

再开始安装之前先检查一下自己的电脑是否具备GPU，如果不具备的话可以使用CPU版本的pytorch。

下面介绍一下如何检查自己电脑是否具有GPU，首先调出任务管理器（Ctrl+Alt+.）,并查看上方菜单栏的‘性能’。

在这里插入图片描述

打开之后查看自己这两个地方是否具有GPU，如果存在的话证明有GPU，不存在说明只能安装CPU版本。

然后在教程中找到相应的GPU和CPU安装版本即可。

在这里插入图片描述

创建虚拟环境

在安装之前我们需要在anaconda下创建一个虚拟环境来装载pytorch.

首先打开‘win’工具栏，在安装好的‘Anaconda’文件夹中，找到‘Anaconda Prompt’并打开。

在这里插入图片描述

打开后输入以下指令，目的是创建一个新的环境(把myenv换成自己想创建的环境名称)。

conda create --name myenv

在这里我创建了一个名为pylpytorch的新环境。

在这里插入图片描述

接下来在弹出的对话框中键入‘y’.

在这里插入图片描述

出现下列对话便是建立完成。

在这里插入图片描述

之后复制粘贴下列代码，目的是切换到新创建的环境（同样把pylpytorch换成你创建的环境名）。

conda activate pylpytorch

此时可以看到从主环境切换到了刚创建的虚拟环境。

在这里插入图片描述

之后复制粘贴下列代码，目的是查看新创建的环境的信息。

conda info

我们可以从中看到新创建的环境相关的各种信息。

在这里插入图片描述

至此环境创建完成。

CPU版本安装

CPU版本的安装相对简单，相较于GPU版本节省了很多步骤，下面我们一步步来安装。

首先附上pytorch官网：https://pytorch.org/

进入之后点击左上角的‘Get Start’。

在这里插入图片描述

进去之后按照箭头指示的地方选中相应的选项。然后复制下图红框中圈到的命令符。

在这里插入图片描述

复制下来后我们从命令符中看到安装pytorch实际上是安装 pytorch torchvision torchaudio这三个库。**由于这个网站里面的命令符下载网站并不在国内，所以直接运用命令指示符下载会非常慢，甚至出现下载失败的情况。**所以建议使用国内镜像下载，下面是一个比较推荐的几个国内镜像网站：

清华镜像：https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

所以此时我们需要把代码更换一下换成：

conda install pytorch torchvision torchaudio cpuonly -c 镜像地址

复制后打开命令指示框，输入上述代码并把镜像地址换成上方清华镜像，如下图：

在这里插入图片描述

后续对话中键入‘y’。

在这里插入图片描述

当出现‘done’如下图的标识时，便是下载成功。

在这里插入图片描述

然后可以通过如下指令来查看是否安装成功

conda list

在这里插入图片描述

输入后回车可以查看环境中已安装的库。会看到一个cpu版本的pytorch。

在这里插入图片描述

这时pytorch已经安装到虚拟环境了，接下来查看库是否可以正常运行。

依次输入python、import torch、torch.cuda.is_available()。出现‘False’说明安装成功并可以正常使用。

在这里插入图片描述

CPU版本pytorch安装至此结束。

GPU版本安装

GPU版本的pytorch相较于CPU版本的pytorch安装略显麻烦，但能够使用GPU的并行计算能力还是很值得的。

安装最新版驱动

首先通过第一步记住自己的GPU型号，比如我的这个NVDIA RTX20系列的2060.

在这里插入图片描述

进入NVDIA下载驱动的网站：https://www.nvidia.cn/Download/index.aspx?lang=cn，然后再产品类型中找到自己的显卡后点击搜索。

在这里插入图片描述

搜索后直接下载安装即可，安装过程比较简单，就不过多赘述了。

在这里插入图片描述

根据自己GPU型号找到合适的cuda版本

安装完成驱动之后，我们需要确定一下适配自己GPU的CUDA版本。

首先使用‘WIN + r’调出命令栏，在命令栏中输入’cmd’。

在这里插入图片描述

在打开的命令框中输入以下代码，目的是查看自己GPU的各种信息以及适配的CUDA版本。

nvidia-smi

如下图可以看到，弹出了自己的GPU及适配信息。其中箭头指向的地方是输入的指令，圈出的地方就是适配的CUDA信息，例如我这边显示的CUDA Version：12.4意思就是我可以安装12.4版本的CUDA版本。

在这里插入图片描述

pytorch安装

首先进入pytorch官网，地址：https://pytorch.org/,点击‘Get Start’。

在这里插入图片描述

情况一：

首先这里列举两种可能遇到的情况，情况一指的是自己的GPU版本不算很靠后，可以在根据本小节教程在Pytorch的官网中找到适配自己GPU的CUDA版本安装指令，如果GPU版本比较靠后在本节找不到对应的CUDA版本安装指令可以移步情况二。

然后再新的页面寻找自己的CUDA版本，例如说我这里适配12.4以下的版本那么我就可以选中蓝色圈出的CUDA 12.1,如下图所示。另外上面四个选项和我箭头段指的选一样的即可，最后复制我红色圈中位置的命令符。

在这里插入图片描述

情况二：

如果从情况一的部分没找到适配自己版本的CUDA版本，我们可以在过去的版本中找，首先在‘Get Start’的页面中找到寻找过去版本的字样。

在这里插入图片描述

进入之后我们使用搜索的快捷键（Ctrl+F）在新的网页中进行搜索我们需要的版本，假如说我需要搜索10.2的CUDA版本，如下图：

在这里插入图片描述

在搜索完成后一定要注意是pip下载还是conda下载，另外也要注意是linux and Windows下载还是osx。

找到适配的命令符后复制即可前往下载安装。

然后打开‘Anaconda Prompt’激活第一开始创建的虚拟环境，例如我创建的是‘pingpytorch’

激活代码为：

conda activate 环境名

在这里插入图片描述

如上图所示，环境切换成功，然后复制在pytorch网站的命令符即可。

但是由于下载源在国外的网站，所以建议使用国内镜像源下载：

清华镜像：https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

在这里插入图片描述

接着在弹出的页面键入’y’即可

在这里插入图片描述

接下来就是等待安装成功。

然后我们确认是否安装成功。依次输入python、import torch、torch.cuda.is_available()。出现‘True’说明安装成功并可以正常使用。

在这里插入图片描述

配置pytorch环境

写到这想起来我的上篇博客有详细的pycharm配置pytorch的步骤，这里我就不过多赘述了，详情直接看上篇博客吧：|Anaconda安装| Anaconda、pycharm安装及配置Python环境详细教程

总结

虽然是把所有步骤都列举出来了，但是安装过程中还是会不可避免的遇到很多问题，不要太影响心态，一点点解决就行，也很欢迎有问题一起讨论。

愿望会实现吧
关注
  • 25
    点赞
  • 踩
  • 17
    收藏
    觉得还不错? 一键收藏
  • 7
    评论
专栏目录
chatgpt赋能python：如何在Python中安装PyTorch
a058046的博客
06-17 186
本文由chatgpt生成，文章没有在chatgpt生成的基础上进行任何的修改。以上只是chatgpt能力的冰山一角。作为通用的Aigc大模型，只是展现它原本的实力。对于颠覆工作方式的ChatGPT，应该选择拥抱而不是抗拒，未来属于“会用”AI的人。🧡AI职场汇报智能办公文案写作效率提升教程 🧡专注于AI+职场+办公方向。下图是课程的整体大纲下图是AI职场汇报智能办公文案写作效率提升教程中用到的ai工具。
windows虚拟环境环境安装pytorch教程，包含虚拟机安装+pytorch安装全流程
03-18
做深度学习或人工智能开发可以使用tensorflow或者pytorch，由于pytorch面向流式编程，对于初学者更加友好，该资源是一份面向初学者的pytorch环境搭建教程，从安装windows虚拟机开始到pytorch安装全流程介绍，...
在GPU上运行pytorch程序（指定单/多显卡)
热门推荐
qq_43307074的博客
11-04 1万+
在GPU上运行pytorch程序（指定单/多显卡)
深入浅出学习Pytorch—Pytorch简介与2024年最新安装（GPU）
xiaoxu的博客
05-03 2313
2017年1月，FAlR（FacebookAl Research）发布PyTorch。PyTorch是在Torch基础上用python语言重新打造的一款深度学习框架Torch是采用Lua语言为接口的机器学习框架，但因Lua语言较为小众，导致Torch知名度不高。
最新最详细的配置PytorchGPU环境
最新发布
qq_49231165的博客
06-06 790
最新最详细的从0开始配置PytorchGPU环境
在英特尔独立显卡上训练ResNet PyTorch模型
英特尔开发人员专区
01-16 7928
本文介绍了在英特尔独立显卡上训练TensorFlow模型的全流程；本文将基于蝰蛇峡谷(Serpent Canyon) 详细介绍如何在英特尔独立显卡上训练ResNet PyTorch模型的全流程。
Python安装Pytorch教程（图文详解）
lvaolan的博客
04-18 877
① Python所有方向的学习路线图，清楚各个方向要学什么东西② 600多节Python课程视频，涵盖必备基础、爬虫和数据分析③ 100多个Python实战案例，含50个超大型项目详解，学习不再是只会理论④ 20款主流手游迫解 爬虫手游逆行迫解教程包⑤ 爬虫与反爬虫攻防教程包，含15个大型网站迫解⑥ 爬虫APP逆向实战教程包，含45项绝密技术详解⑦ 超300本Python电子好书，从入门到高阶应有尽有⑧ 华为出品独家Python漫画教程，手机也能学习。
Windows系统下有英伟达显卡安装PyTorch
算法与编程之美
09-16 1150
1 问题学习深度学习技术需要使用到PyTorch，在安装PyTorch 过程中，有很多的细节需要知道，在下载PyTorch的时候，我们需要根据英伟达显卡的版本进行相应的下载，如何查看英伟达显卡版本？下载Anaconda的时候很慢，如何让Anaconda下载很快？在使用Anaconda下载PyTorch非常慢，如何使用Anaconda下载PyTorch加快？2 方法我们在官网下载Anaconda很慢...
在pytorch中指定显卡
Meiling_up
04-29 251
https://zhuanlan.zhihu.com/p/166161217
Pytorch显卡设置
思念殇千寻的博客
03-01 1825
　　参考资料： 　　https://zhuanlan.zhihu.com/p/166161217 　　本人最常使用到显卡和CUDA的东西莫过于Pytorch了。这篇文章着重说明两个问题：1. 如何import torch并使之输出比较完备的CUDA信息 2. 在服务器上有多张卡的环境下，如何使任务在特定的卡或特定的几张卡上跑。 　　第一个问题： 　　任务目标是输出信息，那么不妨借助Pytorch...
Pytorch安装教程（Python）
S0ybean的博客
07-14 2011
使用Anaconda安装配置Pytorch环境（Python）教程，包含Pycharm无法使用Pytorch包问题的解决办法
Anaconda中创建虚拟环境python3.7并安装pytorch1.4
12-21
切换至清华镜像源 cmd中输入： conda config --add channels ...conda config --set show_channel_urls yes 添加Pytorch 清华源（清华源针对pytorch有单独的源） cmd中输入： conda config --add channels ht
pytorch安装教程，pytorch环境配置
08-03
PyTorch 安装教程和环境配置详解 PyTorch 是一个基于 Python 的机器学习框架，广泛应用于深度学习和人工智能领域。 本文将详细介绍 PyTorch 的安装教程和环境配置，包括 Anaconda 的安装、PyTorch 的安装、GPU 版本...
Python库 | pytorch_forecasting-0.4.1-py3-none-any.whl
04-25
资源分类：Python库 所属语言：Python 使用前提：需要解压 资源全名：pytorch_forecasting-0.4.1-py3-none-any.whl 资源来源：官方 安装方法：https://lanzao.blog.csdn.net/article/details/101784059
Windows系统安装PyTorch框架支持AMD Radeon显卡/Intel显卡
CUFEECR的博客
03-25 2959
PyTorch对NVIDIA显卡的支持最好，但是通过额外配置，也可以支持其他常见显卡，例如通过安装DirectML即可实现使用AMD和Intel显卡，但是性能上可能存在一定的区别，需要根据需要和表现进行灵活选择。
Pytorch本地环境搭建，详细过程
qq_54441735的博客
04-02 1004
最近在学深度学习，帮助同样的深度学习初学者解决Pytorch环境搭建问题，其中包括anaconda的安装，Pycharm的安装，CUDA的下载等部分，包含具体的安装版本，原理解释
关于使用pytorch在30系列显卡（高级别显卡）算力不够问题
门前大橋下丶
11-17 1419
算力
pytorch深度学习环境配置（GPU3070）
ljx12305的博客
04-07 4527
提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档 pytorch深度学习环境配置（GPU3070）前言一、Anaconda的安装二、CUDA安装1.查看显卡2.安装并配置CUDA3.配置CUDA环境变量4.验证一下CUDA是否成功，按win+R 输入cmd三、cuDNN安装四、pytorch安装五、测试pytorch CUDA是否可用总结 前言 拯救者R9000P，NVIDIA3070显卡 一、Anaconda的安装 官网：https://www.anaconda.com/produ.
pytorch在anaconda中安装全过程包括anaconda环境配置
05-11
在Anaconda中安装PyTorch需要以下步骤： 1. 首先下载并安装Anaconda，官网下载地址为：https://www.anaconda.com/products/individual 2. 创建新的虚拟环境（可选）：打开Anaconda Prompt，输入命令`conda create -n myenv python=3.6`，其中`myenv`是自定义的虚拟环境名称，`python=3.6`是选择Python版本，可以根据需要进行修改。 3. 激活虚拟环境：输入命令`conda activate myenv`，其中`myenv`是上一步创建的虚拟环境名称。 4. 安装PyTorch：在Anaconda Prompt中输入以下命令： ``` conda install pytorch torchvision torchaudio cpuonly -c pytorch ``` 上述命令会安装PyTorch及其依赖项，并且指定使用CPU版本。如果要安装GPU版本，可以将`cpuonly`改为对应的GPU版本，例如： ``` conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c conda-forge ``` 其中`cudatoolkit=11.1`是指定CUDA版本号。 5. 安装Jupyter Notebook（可选）：在Anaconda Prompt中输入以下命令： ``` conda install jupyter notebook ``` 安装完成后，输入命令`jupyter notebook`即可启动Jupyter Notebook。 6. 关闭虚拟环境（可选）：在Anaconda Prompt中输入命令`conda deactivate`即可关闭当前虚拟环境。
评论 7
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值