pytorch安装教程

1.Anaconda 下载和安装

下载Anaconda

找到链接: https://www.anaconda.com/products/individual
选择好适合的版本
在这里插入图片描述

安装Anaconda

接下来就是安装了,安装好之后,检查一下是否安装成功

打开Aanconda Prompt
在这里插入图片描述
点击之后进入

在这里插入图片描述
如果可以在弹出的黑色命令行中看到base就意味着安装成功了

2.显卡配置

打开任务管理器
在这里插入图片描述
也许,你之后会遇到不同的项目,需要使用到不同版本的环境。比如这个项目要用到 pytorch 0.4,另一个项目要用到 pytorch 1.0,如果你卸载了0.4版本,安装了1.0版本。那么下一次,你再碰到0.4版本,你就需要卸载1.0版本,安装0.4版本。很折腾。

Anaconda 集成的 conda 包就能够解决这个问题。它可以创造出两个屋子,相互隔离。一个屋子放 0.4 版本,一个屋子放 1.0 版本。你需要哪个版本,就进哪个屋子工作。

我们首先使用 conda 指令创建一个屋子,叫做 pytorch。
输入命令行 conda create -n pytorch python=3.8
之后,弹出提示,输入 y,即可安装。
在这里插入图片描述
安装成功之后,我们要在 pytorch 环境中,安装 PyTorch,(有点绕),使用如下指令,进入 pytorch 环境。
在这里插入图片描述

你可以看到左边的 base 变成了 pytorch,代表成功进入 pytorch 环境。

3.pytorch安装

打开官网并选择合适的版本

在这里插入图片描述
往下拉会发现
在这里插入图片描述
根据自身情况选择合适的版本
第一个推荐选择Stable(1.7.0)
第二个选择操作系统
第三个主要涉及安装方式一般是Conda和Pip,Windows推荐选择Conda,Linux推荐选择Pip
第四个根据安装的python的版本选择
第五个,查看GUP的型号,进行选择

安装pytorc

在这里插入图片描述
在命令行中输入此命令,回车,之后出现很多的包,问你是否要进行安装,果断输入y,之后就是漫长的等待。(我差不多是等了三个小时)
突然安装到84%报错了,我当时就绝望
在这里插入图片描述
根据报错的信息,我删除了几个文件,一次又一次的尝试安装,历经几个小时,终于成功了。

4.验证是否安装成功

(1)在命令行左边为 pytorch 环境中,输入 python
(2)之后,输入 import torch,如果没有报错,意味着 PyTorch 已经顺利安装了
(3)接下来,输入 torch.cuda.is_available,如果是 True,意味着你可以使用 GPU,如果是 False,意味着只能使用CPU。

在这里插入图片描述

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值