Windows下用Anaconda3安装PyTorch环境攻略(针对AMD显卡)

Windows下用Anaconda3安装PyTorch环境攻略(针对AMD显卡)

解决方案goocz2025-03-31 14:24:5512A+A-

近些年来深度学习的发展速度迅猛,从之前谷歌的AlphaGo到如今OpenAI的ChatGPT,一波一波的浪潮席卷着国内的深度学习研究者们。渐渐地,人们开发出了很多款主流深度学习框架。到如今,则基本上是PyTorch和TensorFlow占据了深度学习框架的半壁江山。做深度学习基本上绕不开这两个利器。但是这两个框架功能强大,但是对于小白来说安装配置确是个棘手的问题,笔者接下来就跟大家介绍,如何在AMD显卡的Windows电脑里,用Anaconda3安装PyTorch环境。

一、创建conda环境

默认Anaconda3已经装好(我可能会在之后出一期安装Anaconda3的教程),首先在开始里搜索Anaconda Prompt

根据下述命令创建Python环境

conda create -n pytorch python=3.8.8

根据自己Python的版本进行选择,Python的版本不宜过新。因为很多目前的资料都是用的老版本Python,用最新的Python可能与老版本的语法不一致,会报错。

激活PyTorch环境

conda activate pytorch

二、查看自己的显卡类型

网上的攻略大多是针对电脑使用NVIDIA的显卡的使用者,我这篇是针对使用AMD的用户。大体上相似,安装过程可以借鉴。如果之前不注意自己显卡的类型,误跟着使用NVIDIA显卡的教程走,最后一步一定会报false。

  1. 点击开始,点击运行;
  2. 输入devmgmt.msc,点击确定;
  3. 点击展开显示适配器,下方即为显卡型号;

三、进入PyTorch官网查看下载命令

点击以下链接进入官网

https://pytorch.org/get-started/locally/

因为电脑没有NVIDIA显卡,只能选择CPU作为计算平台,根据以下的红色选项,找到下载命令。

尽量选择稳定版本stable,原因和之前Python版本的选择一样

四、Prompt窗口进行安装

接着在刚才打开的Prompt窗口,在名称为pytorch环境下输入在官网选择的命令

conda install pytorch torchvision torchaudio cpuonly -c pytorch

之后一路yes,就可以看见PyTorch的安装了。

就算用国内那几个大学的源的话,依然会很慢。后续我会讲解下载PyTorch如何提速。

五、完成安装后检测

打开cmd,输入jupyter lab,打开jupyter lab后,再输入以下代码,查看安装效果。

import torch
import torchvision
print(torch)

当输出结果如下时,说明安装已经成功。


好了,安装既然大功告成!那么接下来进行愉快的深度学习吧。

点击这里复制本文地址 以上内容由goocz整理呈现,请务必在转载分享时注明本文地址!如对内容有疑问,请联系我们,谢谢!

果子教程网 © All Rights Reserved.  蜀ICP备2024111239号-5