Ubuntu18.04下Anaconda+pytorch安装

2019-10-28  本文已影响0人  西部的玫瑰

一、搭建pytorch环境

conda create -n pytorch python=3.7

二、安装pytorch

激活新建的pytorch环境

conda activate pytorch

安装最新的pytorch

pip3 install torch===1.3.0 torchvision===0.4.1 -f https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

#从清华镜像源下载安装,经检验十分的快,此处默认是使用 cuda10.1

pip3 install torch==1.3.0+cu92 torchvision==0.4.1+cu92 -f https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

#cuda9.2

pip3 install torch==1.3.0+cpu torchvision==0.4.1+cpu -f https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

#cpu版本

验证输入python 进入

import torch

torch.__version__# 得到结果'1.3.0'

三、配置 Jupyter Notebook

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

conda install ipykernel#安装ipykernel(还是在pytorch 环境下)

python -m ipykernel install --user --name pytorch --display-name Pytorch-for-Deeplearning#写入环境

下一步就是定制 Jupyter Notebook

conda activate base#切换回基础环境

jupyter notebook --generate-config#创建jupyter notebook配置文件,这里会显示创建jupyter_notebook_config.py的具体位置。

打开文件,修改

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

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

至此 pytorch的开发环境安装完成,可以在开始菜单中打开Jupyter Notebook 在New 菜单中创建文件时选择Pytorch-for-Deeplearning 创建pytorch的相关开发环境了,输入如下语句,验证可得即安装成功。

配置pycharm环境即将下图的内核换为envs/pytorch环境下的python内核即可。

如有遇到安装问题即可给我留言。

上一篇下一篇

猜你喜欢

热点阅读