安装要求:
pytorch没有tensorflow那样具有tensorboard可视化工具,在pytorch中想要进行可视化可以调用tensorboardX,具体的调用与tensorboard类似,因此需要的安装包如下:
1、pytorch=1.2、torchvision=0.4.0、cudatoolkit=10.0.130
2、tensorflow-gpu=1.15(有显卡安装gpu版,无显卡安装cpu版)
3、tensorboardX=1.14
上面的版本只是博主自己机器上安装的版本,想要安装其他版本的可以在相应官网上查找对应的版本。
版本对应:
对于gpu版最重要的是cuda版本的选择,因为pytorch和tensorflow都要跟cuda版本对应,选择适合的cuda版本然后到pytorch和tensorflow官网查找对应的版本下载即可,tensorboardX选择1.14是因为默认安装的是2.1版本,调用2.1版本会老是出问题。
pytorch的官网:
/
tensorflow版本对应表:
/install/source_windows
安装顺序:
1、创建conda虚拟环境:
为了便于管理,一般创建conda虚拟环境用于安装各种安装包,安装、激活、关闭、删除环境命令如下:
conda create -n envname python=3.6activate envnamedeactivateconda remove -n envname --all
其中envname为自定义的conda虚拟环境,改为自己想要安装的虚拟环境名称即可,python版本可以改为自己想要的版本。
2、pytorch、torchvision、cuda安装命令:
conda install -n envname pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch
脱坑指南:
<1>直接进入pytorch官网选择对应的版本会显示安装命令,复制到终端安装即可,不用自行安装cuda什么的,上面的指令会直接在对应的虚拟环境下安装对应的cuda;
<2>在官网首页找不到想要的版本可以进入下面的网页查找想要的版本:
/get-started/previous-versions/
<3>conda命令安装pytorch太慢可以在conda添加清华源,再将安装命令中的-c pytorch删掉,因为-c pytorch是从官网下载会比较慢。
3、tensorflow安装命令:
conda install -n envname tensorflow-gpu=1.15
脱坑指南:
有些教程用pip命令安装,博主一开始也用pip命令安装,但老是会出各种各样奇怪的问题,后来全部用conda install -n envname命令安装就没出过问题了,应该是安装包依赖以及安装包位置的问题。
4、tensorflowX安装命令:
conda install -n envname tensorboardX=1.14
脱坑指南:
未指定tensorboardX的版本直接安装会出现各种问题,因为默认安装的是tensorboardX=2.1版本,降为1.14版本就可以了。
测试:
安装完tensorboardX之后,用以下命令测试即可,具体的使用可以查看tensorboardX可视化教程,与tensorboard比较类似。
from tensorboardX import SummaryWriterwriter=SummaryWriter()
如果觉得《conda虚拟环境安装pytorch+tensorboardX可视化工具》对你有帮助,请点赞、收藏,并留下你的观点哦!