Ubuntu18.04 Anaconda环境下配置pytorch(gpu)+tensorboardX

之前在网上看了很多教程,由于好多都过时了,存在不少弯路和坑,因此写一篇比较简便的配置方法。

本文参考了tensorboardX的官方文档,https://github.com/lanpa/tensorboardX

可以看到目前的tensorboardX再以下环境测试过:
*

The current release (v2.1) is tested on anaconda3, with PyTorch 1.5.1
/ torchvision 0.6.1 / tensorboard 2.2.2.

目标是配置 PyTorch 1.5.1 / torchvision0.6.1 / tensorboard2.2.2 +tensorboard2.1

预备工作:安装anaconda,配置清华源

之前看到很多人配置gpu版本的tensorflow或pytorch都要先准备cuda和cudnn,其实可以直接在anaconda中直接配置。

  1. 创建python3.7环境 $ conda create -n 环境名 python=3.7
  2. 激活环境 $ conda activate 环境名
  3. 安numpy(这一步不知道有没有用,但我做了)$ conda install numpy
  4. 配置cuda10.2和cudnn7.6.5 ,$ conda install cudatoolkit==10.2 cudnn==7.6.5
  5. 安装pytorch,这里有一个大坑,如果直接用conda指令安装,会装清华源的pytorch,而清华源的pytorch最多到1.3.0版本,和cuda版本不匹配。
    因此只能用pip安装 $ pip install torch==1.5.1 torchvision==0.6.1 tensorboard==2.2.2
  6. 最后安装tensorboardX ,其实如果只用pytorch,并不需要安装tensorflow
    输入$ pip install tensorboardX,再装soundfile $ pip install soundfile

最后在使用tensorboardX的时候还有一个坑,因此记录一下使用方法
7. 激活环境 $ conda activate 环境名
8. 通过$ tensorboard --logdir "文件路径"进入连接,这里有几个注意的地方,首先必须是文件路径,就是runs的下一级目录,其次必须用双引号“”,不能用等号=。

tensorboard在windows环境下要用等号=,不能用双引号"",而且还得用相对路径!!


2020.10.15
在WIN10上配置环境时遇到好多问题,清华源安装不上cuda和cudnn,用pip安装不上torch1.5.1和0.6.1。
最后用改文方法配置上海交大的源,安装了cuda和cudnn:https://blog.csdn.net/observador/article/details/83618540
然后在pytorch的网站https://download.pytorch.org/whl/torch_stable.html上找到了torch1.5.1和0.6.1的whl文件,下载下来用pip安装的。

你可能感兴趣的:(anaconda,ubuntu,python,cuda)