栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

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

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

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里面的环境和包版本都要一一对应烦死啦!!

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/876083.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号