linux下使用conda安装pytorch,并配置jupyter

首先安装anaconda

  去官网下载.sh包,安装之后会把jupyter一起安装好。
如果没有一起安装,使用下面的命令安装

使用conda
conda install -c conda-forge notebook
或者pip
pip install notebook

配置jupyter

首先先生成一个默认的配置文件

jupyter notebook --generate-config

在~/.jupyter/目录下可以找到一个文件jupyter_notebook_config.py
修改其中几条

c.NotebookApp.notebook_dir = ‘/home/**/jupyter/kernel’ #自定义工作环境目录
c.NotebookApp.ip = '
’ #配置远程访问

最后还需要设置一个密码
进去python环境

from notebook.auth import passwd
passwd()
输入两遍会得到一个sha码
在jupyter_notebook_config.py中修改
c.NotebookApp.password = ‘sha1:**********************************’

这样重启命令行输入jupyter notebook 启动之后,可以在浏览器里输入ip:8888,输入密码访问了。

创建conda环境

conda create -n py37torch python=3.7

为jupyter创建kernel

首先下载conda的ipykernel包,一般安装时已经包含

conda install ipykernel

然后在py37torch环境下安装

conda install -n py37torch ipykernel

然后进入py37torch环境

conda activate py37torch
python -m ipykernel install --user --name py37torch --display-name py37torch

这样在jupyter新建按钮下就会发现一个新的kernel, py37torch

安装torch

去pytorch选择对应的配置和版本会给出安装命令,例如我的python3.7的linux,无cuda版本对应命令如下:

conda install pytorch-cpu torchvision-cpu -c pytorch

验证:
安装好之后在jupyter 中新建一个运行环境,运行以下命令

from future import print_function
import torch
x = torch.empty(5, 3)
print(x)

会输出一个矩阵

安装自动联想功能

在对应的环境下

python -m pip install jupyter_contrib_nbextensions
jupyter contrib nbextension install --user --skip-running-check

或者使用conda

conda install -c conda-forge jupyter_contrib_nbextensions
conda install -c conda-forge jupyter_nbextensions_configurator

重启jupyter在进入的时候会多出一个选项卡nbextensions
勾选table of contents 和 hinterland