2022.11 Windows系统下配置深度学习环境的一些记录(Pytorch,tensorflow1,tensorflow2)

基础环境

安装 CUDA

直接在官网下载安装,依照界面指示即可。

注意,tensorflow1.x的最高版本1.15只支持CUDA10.0
可以在这篇文章中查找CUDA,tensorflow,cuDNN的对应关系。
可以在这里找到所有CUDA版本的下载链接

如果没有条件使用代理,建议先下载好可用的 cuDNN(见下一小节),再下载对应 cuDNN能兼容的 CUDA。

配置cuDNN

此步骤需要注册为Nvidia Developer,注意选择与前一个步骤中 CUDA 版本匹配的版本。官方网站大概率被墙导致网站打不开,可以尝试使用加速器、手机热点、或者其他人分享的内容。

安装anaconda

官网 下载安装。注意默认不会添加 anaconda 到 PATH,而且软件不建议这种做法,但个人经验是勾选能减少很多麻烦。

不要在 base 中安装所有的包,更合适的做法是对同类型的任务,甚至每个任务创建独立的环境。创建环境时建议指定一个不是最新的 python 版本,版本太新太久都将影响兼容性,不指定版本则不会安装 pip,导致后续其他操作非常麻烦。

conda create -n  envs_name python==3.9

安装Pytorch

Pytorch 安装较为简单,进入官网 根据系统和 CUDA 选择合适的版本后,复制下方的 command,在anaconda 对应的 evns 中执行即可。

安装偶尔会存在网络连接不稳定的情况,有条件的建议全程使用代理。

安装tensorflow

tensorflow2.x

tensorflow2.x 可以直接使用pip安装,默认的 pip install tensorflow 就是对应最新2.x版本。

tensorflow1.x

tensorflow1.x的安装需要显式的指定tensorflow版本,并指定是否为gpu版本。

注意版本号的格式为A.B.C,只输入前两部分可能提示找不到对应版本

pip install --upgrade tensorflow==1.15.0  # 仅支持 CPU 的版本
pip install --upgrade tensorflow-gpu==1.15.0  # 仅支持 GPU 的版本

如果 Python版本不合适,将提示没有对应的版本。注意对应环境中应安装合适的 Python 版本,对于tensorflow==1.15,实测 Python3.7 或3.8均可成功安装。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值