之前在网上看了很多教程,由于好多都过时了,存在不少弯路和坑,因此写一篇比较简便的配置方法。
本文参考了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中直接配置。
- 创建python3.7环境
$ conda create -n 环境名 python=3.7
- 激活环境
$ conda activate 环境名
- 安numpy(这一步不知道有没有用,但我做了)
$ conda install numpy
- 配置cuda10.2和cudnn7.6.5 ,
$ conda install cudatoolkit==10.2 cudnn==7.6.5
- 安装pytorch,这里有一个大坑,如果直接用conda指令安装,会装清华源的pytorch,而清华源的pytorch最多到1.3.0版本,和cuda版本不匹配。
因此只能用pip安装$ pip install torch==1.5.1 torchvision==0.6.1 tensorboard==2.2.2
- 最后安装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安装的。