Anaconda下的pytorch环境配置

目录

  • 1 安装cuda和cudnn
  • 2 安装Anaconda
  • 3 安装pytorch
    • 3.1 添加清华源
    • 3.2 创建pytorch环境
    • 3.3 激活pytorch环境
    • 3.4 安装pytorch
    • 3.5 测试安装是否成功
      • 3.5.1 命令行
      • 3.5.2 jupyter notebook
        • 导入torch

由于第1步和第2步比较简单,所以直接给出参考博客。

1 安装cuda和cudnn

参考此博客: cuda安装教程+cudnn安装教程.
如果没有GPU的话,忽略这一步。

2 安装Anaconda

在Anaconda官网下载Individual Edition
如下图:Anaconda下的pytorch环境配置_第1张图片

图1 Anaconda下载界面

我是按照默认选项装的。
具体安装步骤可参考:Anaconda的下载和安装步骤

3 安装pytorch

3.1 添加清华源

在Anaconda Prompt中分别输入以下四句代码:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

3.2 创建pytorch环境

conda create -n pytorch python=3.7

可以依此类推创建自己需要的环境

conda create -n env_name python=x.x(3.7)

相当于为pytorch新建一个小房间。如果不进行这一步的话,就很有可能把pytorch安装在base环境下,虽然也不会报错,但很容易把环境弄乱。
注意
有些环境需要配置在指定的python版本下,这里我们按照python3.7来配置pytorch即可

3.3 激活pytorch环境

先查看上一步创建的pytorch环境是否成功

conda env list

或者

conda info --envs

上述两行命令的作用都是查看环境,如果出现如下图所示的pytorch环境,说明pytorch环境创建成功
Anaconda下的pytorch环境配置_第2张图片

图2 查看环境

接着用下面这条命令激活pytorch环境

conda activate pytorch

激活成功后,图2括号中的(base)将变为(pytorch)如图3
Anaconda下的pytorch环境配置_第3张图片

图3 激活pytorch环境

完成此步之后才可继续进行下一步安装pytorch,否则pytorch将被装在base环境下。

3.4 安装pytorch

在PyTorch官网.根据自己的配置,如下图4,复制对应的命令。
Anaconda下的pytorch环境配置_第4张图片

图4 pytorch安装

如果没有GPU,Compute Platform选择CPU
复制下面这一行命令,在3.3的基础上运行这一条命令

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

安装的过程中会弹出提示[y/N]?
选择y即可,等待安装完成

3.5 测试安装是否成功

3.5.1 命令行

你可以选择在命令行下依次输入以下代码

python

在这里插入图片描述

import torch

Anaconda下的pytorch环境配置_第5张图片
如上图,没有报错,说明pytorch环境配置成功
还可以打印出pytorch的版本

print(torch.__version__)

Anaconda下的pytorch环境配置_第6张图片

3.5.2 jupyter notebook

也可以在jupyter notebook界面下导入torch,检查环境是否配置成功。但是在这一步中,我曾经遇到过以下问题,(不知道是不是所有人都会有:( )

No module named 'torch'

原因可能是anaconda默认只在base环境下装了jupyter notebook,尽管已经激活的pytorch环境,具体原因我目前还解释不上来,给自己留个坑以后回来补吧。
一种解决方法是在Anaconda Navigator主界面选择pytorch环境,安装jupyter notebook
Anaconda下的pytorch环境配置_第7张图片
安装完之后可以在这里直接打开jupyter notebook,但我比较喜欢在命令行下打开。
在命令行下输入

jupyter notebook

即可打开jupyter notebook,但默认打开的是根目录,如果你的.ipynb文件在其他文件夹的话,可以参考一下方式打开jupyter notebook。

  1. 复制文件夹路径
    在这里插入图片描述
  2. 命令行内转到对应的文件夹
cd /d H:\DeepLearning\机器学习xxx

在这里插入图片描述
再输入jupyter notebook,将在默认浏览器中打开jupyter notebook
Anaconda下的pytorch环境配置_第8张图片
如果没有打开,复制链接到浏览器中即可。
Anaconda下的pytorch环境配置_第9张图片

导入torch

在这里插入图片描述
说明pytorch环境配置成功,可以在jupyter notebook上使用torch包啦。

你可能感兴趣的:(深度学习,python,深度学习,pytorch,anaconda)