pytorch 环境搭建

一、安装Anaconda。 根据需求,下载相应的版本安装即可。

这里写图片描述

二、登陆至pytorch官网,根据环境,选择不同的安装语句。
这里写图片描述

以Linux,cuda8.0,conda环境 为例,运行以下语句

conda install pytorch torchvision -c pytorch

三、安装tensorflow,参考链接

GPU版本

conda install -c anaconda tensorflow-gpu

CPU 版本

conda install -c anaconda tensorflow

四、安装tensorboardX

pip install tensorboardX

或者

pip install git+https://github.com/lanpa/tensorboard-pytorch
阅读更多
文章标签: pytorch install
个人分类: pytorch
所属专栏: pytorch学习笔记
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭
关闭