一、选择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下载命令
二、创建虚拟环境
如果你还没有创建一个虚拟环境,打开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则表示安装成功