pytorch安装
用conda
conda install pytorch torchvision -c pytorch
用pip (我用这个可以)
pip install torch torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple
tensorboardX安装
用pip安装
pip install tensorboardX
从源码安装 (我用这个可以)
git clone https://github.com/lanpa/tensorboardX && cd tensorboardX && python setup.py install
|