(2023)Anaconda+GPU环境之下Pytorch安装教程。

一、说明

本文在安装Pytorch之前已默认安装好Anaconda,若未安装好,可以寻找别的教程进行安装(Anaconda),此外,本文将讲解如何查看自己电脑是否配备有GPU,基于以上基础,进行Anaconda+GPU环境之下Pytorch安装。

二、查看电脑是否有GPU

点击开始(右键单击)-- 任务管理器 --  点击性能 --查看电脑是否有GPU

如下图中,则显示我自己的电脑有GPU,若没有GPU,则可以基于CPU(蓝色框显示自己电脑的CPU)进行安装。

三、使用conda命令下载安装Pytorch之准备工作+虚拟环境的创建及激活。

1.首先打开Anaconda之下的Anaconda Prompt进入命令行。

2.在命令行中输入python来查看自己的python版本,在此显示我的版本是3.11.5。查看完python版本之后,输入exit()点击回车来退出python环境。

3.使用conda命令来创建虚拟环境,之后将pytorch安装在此创建的虚拟环境之中。

conda create -n 虚拟环境名字 python=版本

例如:在此,我将基于python3.11.5建一个虚拟环境名称为pytorch的虚拟环境

conda create -n pytorch python=3.11.5

在虚拟环境的创建之中,如遇Proceed ([y]/n)? 则输入y,表示继续安装。

安装完成之后,输入以下代码显示目前所有的虚拟环境

conda info --envs

下图显示我的电脑在此环境之下的虚拟环境,出现了刚才安装的pytorch虚拟环境,则表示安装成功。

4.使用conda命令激活虚拟环境,进入刚才创建的pytorch虚拟环境。

conda activate 虚拟环境名字

例如,激活刚才创建的pytorch虚拟环境则输入:

conda activate pytorch

四、使用conda命令下载安装pytorch

说明:此处进入pytorch官网,采用pytorch官网所提供的conda命令进行pytorch安装。

1.首先确定自己电脑的显卡型号,以确保在安装pytorch时可以选择合适的cuda版本。

在命令行中输入nvidia-smi,按回车,则会显示自己电脑的CUDA版本。在以下图中,则显示我电脑的CUDA版本是12.2。

2.进入pytorch官网,选择自己的版本型号,复制粘贴代码进入命令行中进行安装。(注:为了提高下载速度,之前我选用其他博主所提供的镜像源方式进行安装,但最后验证的时候出错了,最终发现使用镜像源所下载的pytorch是基于CPU所安装的,而我所安装的是基于GPU进行的,故在此还是选择这种官网下载的方式进行,静心等待吧,速度慢点也没关系。)

以下展示官网下载pytorch的一些选择。

如遇Proceed ([y]/n)? 则输入y,表示继续安装。

3.下载安装完成之后,仍然在我们创建的pytorch虚拟环境之中,输入conda list查看是否有pytorch或者torch。如下图显示有pytorch,则代表安装成功。

conda list

输入python,再输入import torch,最后输入torch.cuda.is_available(),若显示True,则代表基于Anaconda+GPU环境之下的pytorch安装成功。(注意:若安装成功,但输入此语句显示False,则原因之一是因为你安装的是基于CPU环境之下的,还需进一步检查。)

import torch

torch.cuda.is_available()

至此,基于Anaconda+GPU环境之下的pytorch安装成功。若有啥问题和建议或我的文章的不足,可以在留言区留言,我看到了都会回答的。后续我也会更新一些安装此过程中存在的一些问题,以及根据自己电脑nvdia版本来安装和更新显卡教程。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值