CUDA11.0 如何配置下载Pytorch1.0.0

挣扎了两个下午

参考文章

1.Windows下安装PyTorch1.0_nudt_qxx的博客-CSDN博客_windows安装pytorch

2.WIN10搭建深度学习环境（Pytorch1.0.0+Python3.7+Anaconda5.3.1+CUDA10.0+Pycharm）_读博好难啊的博客-CSDN博客_windows10 搭建深度学习环境

联邦学习目前学习Model-Contrastive Federated Learning这篇论文要求环境为

• PyTorch >= 1.0.0
• torchvision >= 0.2.1
• scikit-learn >= 0.23.1

一定要看代码里的requirements.txt ,github写的不对！

其实真正的要求是

torch==1.0.0
torchvision==0.2.1
scikit-learn>=0.23.2

故为pycharm搭建pytorch环境

1.检查电脑CUDA版本 我的版本为11.0 似乎PyTorch1.0.0只支持8.0 9.0 10.0

故参考博文2下载CUDA 10.0

2.在cmd中 activate你的环境 参考博文1进行pip安装就行了

pip install https://download.pytorch.org/whl/cu100/torch-1.0.0-cp37-cp37m-win_amd64.whl

3.检查是否安装成功

pip list

另外一个安装包超级好用的命令

pip install -i https://pypi.douban.com/simple/ scipy==版本号(scipy 替换为你想安装的包)

eg：pip install -i https://pypi.douban.com/simple/ torchvision == 0.2.1

卸载命令为

pip uninstall 

Pycharm里面的环境和包版本都要一一对应烦死啦！！