本来的环境是tensorflow GPU 2.0的,下了个pytorch的代码要跑,所以要装pytorch。
pytorch安装就容易的多了,官网有根据系统和CUDA版本给的安装版本,选择用pip进行安装(过程发生了很多麻烦事儿)
# CUDA 10.0
pip install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html
# CUDA 9.2
pip install torch==1.2.0+cu92 torchvision==0.4.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html
# CPU only
pip install torch==1.2.0+cpu torchvision==0.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
不知道是不是下载速度慢的原因,两次都没有成功,然后有警告pip有更新版本,就想着更新了呗,然后悲催的更新失败,pip竟然不见了。报错:No module named pip
百度了下最常见的方法是:
python -m ensurepip
easy_install pip
好巧不巧No module named ensurepip ,重启啥的也没有解决这个问题,有说法讲是环境变量配的不对,然后我看了看我python/Scripts的路径确实是错了,但奇怪的是,tensorflow跑Mask R-CNN一直都是可以的。我把scripts的路径改正之后,重启依旧没有pip和ensurepip,单独跑了
easy_install pip
然后pip在下载安装,我也不知道是怎么个情况,反正pip就能用了,pytorch也装上了。
然后安装tqdm,进度条
pip install tqdm