PyTorch安装教程【小白入门】

本文详细指导了如何在Windows系统上为深度学习项目安装PyTorch和CUDA,包括选择合适的PyTorch版本、创建并激活Anaconda虚拟环境、下载并安装PyTorch,以及检查安装是否成功。适合初学者和经验丰富的开发者参考。
摘要由CSDN通过智能技术生成
一、选择PyTorch版本

上期文章解决了如何安装anaconda和CUDA,这期就来解决PyTrch的安装问题,首先根据安装的CUDA版本来选择PyTorch的版本,如果没有安装CUDA的,可以去看我上一期文章

CUDA安装教程(包含版本选择,安装失败,环境变量配置)-CSDN博客文章浏览阅读451次,点赞3次,收藏2次。本文提供了一个详细的指南,帮助深度学习开发者在Windows系统上快速安装PyTorch和CUDA。我们将逐步介绍如何设置Anaconda虚拟环境、安装指定版本的PyTorch以及配置CUDA,以确保你的深度学习项目可以充分利用GPU加速。无论你是刚入门的AI爱好者还是有经验的机器学习工程师,这篇文章都将为你提供所需的一切信息,让你的开发环境配置变得轻松而无忧。https://blog.csdn.net/weixin_52443865/article/details/136374938?spm=1001.2014.3001.5502
CUDA安装完后,打开以下网址,找到对应cuda版本的PyTorch下载命令

Previous PyTorch Versions | PyTorchInstalling previous versions of PyTorchicon-default.png?t=N7T8https://pytorch.org/get-started/previous-versions/

二、创建虚拟环境

如果你还没有创建一个虚拟环境,打开cmd终端,一条一条输入运行以下命令创建环境

conda create -n your_env_name python=3.7

conda activate your_env_name

conda env list

【记得将your_env_name替换成你要创建的环境名字,如:envpython3.7】

运行完最后一个命令后,显示出你的虚拟环境后便创建成功


进入虚拟环境

输入conda activate your_env_name或者activate your_env_name【有时第一个会报错,使用第二个即可】,进入环境

三、下载安装PyTorch

开始下载安装pytorch,输入之前准备好的命令

最好选后面带下载地址的,博主之前用没有地址的总是卡住。
有可能会出现下载错误和异常中断的问题,只要重复运行这个下载命令,然后等完成安装完成即可

重复安装会读取之前的下载好的缓存,只会下载未完成的部分

安装成功

四、检查是否安装成功

输入conda list

出现torch或pytorch则安装成功

测试,输入以下语句,显示版本和返回true则表示安装成功
 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值