最近机器学习的课设需要用到Tensorflow来跑代码。
有CPU和GPU两个版本的Tensorflow(Google怎么这么优秀,气哭.jpg)
电脑是GTX1050Ti的显卡,4G。系统是Win10的。
上网查安装GPU版本的Tensorflow要有CUDA和cuDNN的支持,于是去看CUDA和cuDNN的安装教程(安软件安到害怕的我这次选择先仔细看网友们的填坑总结,再开始安装,事实证明真是个正确的选择!)
网上的安装博文很多,这里就不再重复了,还要去看数模论文,时间紧迫,先记录一下要紧的部分。
千坑万坑!最恶心的是!版本问题的坑!
总结了很多博文,得出了一个避坑方法:
- 如果你是为了Tensorflow安装的这两个东西,那么你应该先去Tensorflow的官网查看一下安装教程(看GPU版本的那个!)仔细看看,现在的Tensorflow支持哪个版本的CUDA。
- 确定支持的CUDA版本之后,你需要去CUDA的官网,看一下——你的显卡版本支不支持CUDA!(简直是个神坑!)当然,大部分最新的显卡都是支持的,这里提一下:如果你的显卡跟我的一样,是GTX1050Ti的,那么虽然官网上没写支持,你也可以安装(亲测有效!)。所以这个官网真的很奇怪,但保守一点还是去看一眼确定一下为妙。
- 在CUDA的官网下载相应的Tensorflow支持的版本,安装即可(详细步骤看网友们的其他博文)
- 安装cuDNN:这个时候你又要看了,你下载的这个CUDA支持哪个版本的cuDNN!(这个就比较好了,官网在每个后面都注明了每个cuDNN支持哪个版本的CUDA,非常人性化,如果你找不到你下载的版本,那说明可能是个老版本,你可以去Release字样的超链接网页里面找一下)
到这里,CUDA和cuDNN的版本大坑就已经成功的跨过去啦!其余的安装步骤、环境配置步骤,按照网友们博文中说的来就好啦!
PS:记住!网友们的电脑、系统等等等等可能跟你自己的都不一样的!所以下载这两个东西时不能完全按照网友们的博客来!我上面总结的是通用的办法,再结合具体的安装方法就完全可以成功安装!希望能帮助到大家~
去看数模了,886~