如何安装虚拟环境(深度学习环境GPU搭建)

如何安装虚拟环境(深度学习环境GPU搭建)

安装tensorflow(cpu/gpu)、pytorch(cpu/gpu)时,个人踩坑的点
在linux和windows下均可使用
有两种方法
第一种:基于安装NVIDIA驱动的,先百度显卡支持的cuda以及cudnn的版本。然后按照正常情况安装驱动,再搭建匹配的深度学习环境即可。(Windows与Linux都可行)
优点:不需要额外安装cuda、cudnn的一些包
缺点:在显卡驱动更新或者linux内核更新的情况下,搭建的深度学习环境可能会出现不能使用的情况。因此,第一种方法适用于版本不常更新的情况。
第二中:使用anaconda安装虚拟环境,安装好anaconda后,新建一个虚拟环境,然后在虚拟环境里面使用conda命令安装需要的tensorflow-gpu版本,anaconda会自动安装好相应的一些cuda、cudnn版本。同理pytorch-gpu的话直接conda执行官网的指令(也可pip执行指令)。(Windows与Linux都可行)
优点:可安装多个虚拟环境,无需考虑显卡的一些更新版本问题。环境之间互相没影响。
缺点:安装缓慢(可自行换清华源或者离线安装)安装多个虚拟环境的话所占空间较大。

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值