Windows下Anaconda中安装pytorch

r囧r小猫 2023-01-08 08:19 206阅读 0赞

## 前提 ##

在Anaconda Prompt中建立新环境（python3.6版本的）建立新环境的目的，是为了适应不同版本的切换，比如，你想使用3.6版本的python，就切换到3.6的。使用3.7的就切换到3.7的。

conda create -n 当前环境名称 python=3.6

安装好后，会有提示：  
To activate this environment, use（激活环境）

#  #

conda activate 环境名称

#  #

To deactivate an active environment, use（退出环境）

conda deactivate

## 安装pytorch ##

一、首先打开Anaconda Prompt

二、创建一个python3.6的虚拟环境

conda create -n pytorch python=3.6

三、进入虚拟环境

conda activate pytorch

四、虚拟环境下执行（我使用的是cpu版本的）

conda install pytorch torchvision torchaudio cpuonly -c pytorch

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FzMTc4MDkyNTk3MTU_size_16_color_FFFFFF_t_70]

这个命令是到[pytorch官网][pytorch]中，根据你想要得gpu或者cpu版本去复制。一般，推荐liunx使用pip，windows使用conda。

如果选择gpu版本的话，要先看看自己的显卡型号，然后选择对应的cuda。推荐看一下[B站视频][B]。  
五、检查是否安装成功

pip list

如下图，可以看见，已经安装成功pytorch1.7.1  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FzMTc4MDkyNTk3MTU_size_16_color_FFFFFF_t_70 1]  
六、导入使用

python

进入python环境后，使用

import torch

导入torch，如果不报错，说明安装成功。  
然后在执行,判断是否可用gpu

torch.cuda.is_available()

显示False是不可用（因为我安装是cpu版本的），True是可用。  
![在这里插入图片描述][20210119160647412.png]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FzMTc4MDkyNTk3MTU_size_16_color_FFFFFF_t_70]: /images/20221119/be0db8fc4e6245eaa8b162d21d5f9264.png
[pytorch]: https://pytorch.org/
[B]: https://www.bilibili.com/video/BV1nE411b7Vx?from=search&seid=15455831906003568026
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FzMTc4MDkyNTk3MTU_size_16_color_FFFFFF_t_70 1]: /images/20221119/d0541363288947d095d2b595f1995b3c.png
[20210119160647412.png]: /images/20221119/01bccbccb5e3422681878c98386927fc.png