Pytorch环境搭建

Pytorch 环境搭建

PyTorch 的安装十分简单，根据 PyTorch 官网，对系统选择和安装方式等灵活选择即可。这里以 anaconda 为例，简单的说一下步骤和要点。国内安装 anaconda 建议使用清华或者中科大 [http://mirrors.ustc.edu.cn/help/anaconda.html] 镜像，快的不是一点半点。

1.安装 Pytorch

anaconda 安装完成后可以开始创建环境，这里以 win10 系统为例。打开 Anaconda Prompt

#pytorch 为环境名，这里创建 python3.6 版。
conda create -n pytorch python=3.6

# 切换到 pytorch 环境
activate pytorch

#安装 GPU 版本，根据 cuda 版本选择 cuda80，cuda92，如果 cuda 是 9.0 版，则不需要
#直接 conda install pytorch -c pytorch 即可
# win 下查看 cuda 版本命令 nvcc -V
conda install pytorch cudatoolkit=10.0 -c pytorch

# cpu 版本使用
# conda install pytorch-cpu -c pytorch
# torchvision 是 torch 提供的计算机视觉工具包，后面介绍

pip install torchvision

需要说明的一点是如果使用清华源，可以直接添加 pytorch 源镜像去掉，并且去掉 - c pytorch 这样才能使用镜像源。

验证输入 python 进入

import torch

torch.__version__

# 得到结果 '1.0.1'

2. 配置 Jupyter Notebook

新建的环境是没有安装安装 ipykernel 的所以无法注册到 Jupyter Notebook 中，所以先要准备下环境

#安装 ipykernel
conda install ipykernel

#写入环境
python -m ipykernel install  --name pytorch --display-name "Pytorch for Deeplearning"
#如果报无权限的错误，改为以下命令
sudo /home/henry/anaconda3/envs/pytorch/bin/python -m ipykernel install --name pytorch --display-name "Pytorch for Deeplearning

下一步就是定制 Jupyter Notebook

#切换回基础环境
activate base

#创建 jupyter notebook 配置文件
jupyter notebook --generate-config

## 这里会显示创建 jupyter_notebook_config.py 的具体位置

打开文件，修改

c.NotebookApp.notebook_dir = '' 默认目录位置

c.NotebookApp.iopub_data_rate_limit = 100000000 这个改大一些否则有可能报错

Pytorch环境搭建

Comments