挣扎了两个下午
参考文章
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.0torchvision >= 0.2.1scikit-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里面的环境和包版本都要一一对应烦死啦!!
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)