conda虚拟环境安装pytorch+tensorboardX可视化工具

最新推荐文章于 2024-08-17 11:46:04 发布

StarCap

最新推荐文章于 2024-08-17 11:46:04 发布

阅读量4.3k

点赞数 1

分类专栏：那些年踩过的坑 tensorboardX安装 pytorch+tensorflow安装文章标签： anaconda 深度学习 python 可视化

本文链接：https://blog.csdn.net/StarCap/article/details/113596545

版权

本文介绍了如何在conda虚拟环境中安装PyTorch 1.2、Tensorvision 0.4.0、CUDA 10.0.130以及TensorboardX 1.14。详细步骤包括创建conda环境、安装PyTorch、TensorFlow-GPU（根据是否有GPU选择版本）、TensorboardX，并提供了安装过程中的注意事项和解决常见问题的建议。

摘要由CSDN通过智能技术生成

安装要求：

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的官网：