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

深度学习搭建环境的坑

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

深度学习搭建环境的坑

  • 使用conda来管理环境
  • conda和pip安装库可以混用,但一般选pip,因为pip的官网库更多更全,下载更快
  • 尽量安装pytorch GPU版本,TensorFlow GPU版本
  • pip install torch1.7.0+cu101 torchvision0.8.1+cu101 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple

-模型训练,batchsize从个位数开始,大了训练报错

  • pytorch和TensorFlow不要安装最新版本,切记,坑多,一般安装版本列表里面居中的版本
  • 装环境时尽量在服务器上,也不需要去装cudakit
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/870466.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

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

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