anaconda下安装pytorch

目录

1.安装anaconda，推荐：

2.安装pytorch 

    2.1 安装cuda，此处以cuda10.1版本为例

   2.2 安装cudnn

   2.3 创建pytorch环境

   2.4 安装pytorch

---

1.安装anaconda，推荐：

https://blog.csdn.net/wq_ocean_/article/details/103889237?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522162639509816780264075304%2522%252C%2522scm%2522%253A%252220140713.130102334..%2522%257D&request_id=162639509816780264075304&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~all~top_positive~default-1-103889237.first_rank_v2_pc_rank_v29&utm_term=anaconda安装教程&spm=1018.2226.3001.4187

---

2.安装pytorch 

    2.1 安装cuda，此处以cuda10.1版本为例

下载链接：CUDA Toolkit Archive | NVIDIA Developer

下载完成后，点击安装。

选择自定义安装。其他的都是不用改，一直往下就行了。

 下载完成后，进入高级系统设置，找到环境变量

 

 

 测试是否成功：

---

   2.2 安装cudnn

 官网下载：https://developer.nvidia.com/rdp/cudnn-archive

去官网下载需要注册后才能下载。

 下载完成后解压。

然后找到cuda的安装路径，在安装时有显示路径，记得记下来。

分别将cuDNN三个文件夹的内容分别复制到cuda对应的文件夹里面。如下所示：

 检查是否安装成功：

Windows10检查Cuda和cuDNN是否安装成功？_jhsignal的博客-CSDN博客

---

  2.3 创建pytorch环境

（1）创建环境pytorch，使用Python版本是3.7

conda create -n pytorch python=3.7

之后，弹出提示，输入 y，即可安装。

（2）查看环境是否安装成功

conda info --envs

可以显示出pytorch环境

---

  2.4 安装pytorch

进入网站：PyTorch

根据自己的版本选择对应的语句： 

点这里进入之前的版本，找到对应的命令语句

 

 

对应CUDA 10.1的语句有很多个，选一个就行。

然后打开cmd

激活虚拟环境 ：

activate pytorch

在虚拟环境下输入找到的命令 ：

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

 弹出提示，输入 y，即可完成安装，显示“done”。

测试：

（1）进入python：python

（2）测试numpy：import numpy

（3）测试pytorch：import torch