Pytorch(cpu+Windows10+Anaconda(Python3.7)+Pycharm+Jupyter Notebook+清华镜像源) 环境搭建

目录搭建Pytorch1.4.0环境安装 Pytorch1.4.0在Pycharm中加入解释器在Jupyter Notebook中修改kernel
搭建Pytorch1.4.0环境

希望本文能帮大家解决一些问题,避开一些坑

安装 Pytorch1.4.0

在Anaconda中创建环境

conda create --name pytorch python=3.7 anaconda
conda activate pytorch #创建完成后激活该环境

添加清华镜像源channel

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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

加下来安装pytorch包。注意,主流方法是在pytorch官网上查询对应的pip代码,但由于我们这里要使用镜像源,所以代码会有一点不同。

pip install pytorch-cpu torchvision-cpu

博主这里安装的是 Windows10 + conda + cpu版本
注意,如果是在官网上查到的代码,在后面还会有一个”-c pytorch“的小尾巴,我们要把它去掉,不然就不是使用镜像源下载了。

安装好了以后,我们进行基本的测试

import torch
import torchvision
print('torch.__version__')

输出为

1.4.0

说明安装成功。

至此,大部分的任务算是完成了,但是大部分人需要在 PycharmJupyter Notebook中使用Pytorch。
我们接下来介绍怎么讲我们打建好的Pytorch环境导入到这两个中去。

在Pycharm中加入解释器

打开Pycharm,在 settings 里选择 Project Interpreter
点击 Add
注意是在anaconda3的env文件夹下面,找到对应的环境名(pytorch),再选择这个目录下的python.exe作为解释器。
注意是在anaconda3的env文件夹下面,找到对应的环境名(pytorch),再选择这个目录下的python.exe作为解释器。

完成:)
你可以再尝试刚刚的三行测试代码,结果不报错,则说明设置成功。

在Jupyter Notebook中修改kernel

这个非常关键,很多朋友可能刚打开了notebook,就遇到 “No Module named torch” 的问题。

首先进入 Anaconda Navigator,在 Environment 中找到我们刚刚搭建好的名为 pytorch 的环境,安装以下两个package:

nb_conda
nb_conda_kernels

在这里插入图片描述

这两个包非常重要,它可以让你在notebook中自由选择你所配制的环境的kernel

安装成功后,在我们的pytorch环境下启动jupyter notebook

conda activate pytorch
cd *指定ipynb文件的目录文件夹*
jupyter notebook #启动

接下来是关键步骤
在Kernel处,把kenel修改成我们自己建的pytorch环境对应的kernel
在这里插入图片描述
接下来就可以跑袋麻辣(跑代码啦)!
希望大家安装顺利!(笔芯

作者:克里斯桃的大胡子

相关推荐

Django cookie 与 session

Django cookie 与 session

Django ORM – 多表实例

Django ORM – 多表实例

Django ORM – 多表实例(聚合与分组查询)

Django Form 组件