python新手——安装tensorflow-gpu过程得到的一些启示

安装tensorflow-gpu的过程中遇到了很多问题,也有一些感悟,所以记录一下,想让大家少踩点坑。其实就是注意一些需要注意的点,这样在安装的过程中就不至于一头雾水了。

1. 虚拟环境

看到官方教程上需要先创建虚拟环境,那为什么要创建虚拟环境呢?

因为python版本很多,不同的包版本也不同,全部在一个大环境下安装就很混乱了。你可以当虚拟环境就是一个隔离的空间,如果你刚创建一个虚拟环境,这里就完全是新的空白的空间,不用管之前装了什么东西,在这里重新弄就完事了~ 当然这样需要的存储空间也大了,因为在不同的虚拟环境里你可能要重复装同样的包,不过我觉得值得。

虚拟环境创建、删除,可参考https://blog.csdn.net/sizhi_xht/article/details/80964099

2. 注意版本!

安装gpu版本的tensorflow会比较复杂,因为还要装对应的cuda,cudnn这些包。
可能会感觉各个版本啥的很乱,其实稍微理一下就会清晰很多。

1. 先看看官网 https://www.tensorflow.org/install/source_windows

这里有不同tensorflow-gpu版本对应的python版本,cudnn版本,以及cuda版本。其实就注意这三个就好了。(还没正式开始,先留个印象)

2. 看你电脑的显卡型号

如何看显卡型号可参考:https://jingyan.baidu.com/article/e73e26c06ef2b624acb6a743.html

OK!知道显卡型号后就可以正式开始了。

  1. 通过显卡型号确认对应的 CUDA 版本
  2. 确定 CUDA 版本后 就可以确定匹配的 cudnn 版本了
  3. 回过头来看第一点,确认你的tensorflow-gpu版本
  4. 然后确认你的python版本

这样是不是对你要安装的包的版本就都清楚了。
注意!创建虚拟环境时就需要指定python版本!

这里有一个Windows安装GPU版本的tensorflow的教程挺详细的,我最初是按照这个来的(但是没完全按里面的做,出了些版本上的问题。里面卸载显卡那块我试了,电脑重启了,重启后就界面就有点难看了,让我慌了一阵,不过后面又好了。我觉得可以跳过这一步试试,如果有报错再好好探究一下吧~省的心慌)
https://www.cnblogs.com/guoyaohua/p/9265268.html

也可以参考其他的安装教程,但我觉得很重要的一点就是先明白你要装什么版本。当然可能还会有其他的坑。其实回想起来过程也不复杂。望安装顺利!: )

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值