配置jupyter-pytorch深度学习环境

配置jupyter-pytorch深度学习环境

• 安装anaconda3
• 新建环境，命名为pytorch
• 在虚拟环境里安装jupyter

activate pytorch
pip install jupyter

• 安装可视化插件，ipywidgets，并且关联

pip install ipywidgets
jupyter nbextension enable --py widgetsnbextension

• 配置jupyter里面的虚拟环境

activate pytorch
pip install ipykernel
python -m ipykernel install --name pytorch

• 安装torch(cuda版本自己对应)

conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch

• 在pytorch中使用tensorboard

pip install tensorboardX
pip install tensorflow #在pytorch环境中安装一个tensorflow,这是个坑，别忘了装

from tensorboardX import summaryWriter
writer = summaryWriter(log_dir="你的存储log的目录")
#这里，在writer里增加需要查看的数据
writer.close()

#打开终端
tensorboard --log_dir "你的存储log的目录"