pytorch 版本使用1.1.0 参照下列命令
pip3 install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp36-cp36m-win_amd64.whl ##(CUDA 10.0) https://download.pytorch.org/whl/cu90/torch-1.1.0-cp36-cp36m-win_amd64.whl ##(CUDA 9.0)
和tensorboardx使用 1.8版本
pip install tensorboardX==1.8
安装前 卸载当前的torch 和tensorboardx
其实有很多坑,注意!
尽量少动环境