win10 布置深度学习 2021.1

win10 布置深度学习 2021.1

几经摸索,参照其他,总结一下经验,以备将来重装

一、安装tensorflow.-gpu
1、安装anacoda3 此步一定要选对版本,最新版的是python3.8,在windows上安装tensorflow-gpu基本不会成功,网上早期教程让用python3.5版本,经测试现在python3.7版本也已通过
2、安装n卡驱动,驱动版本应该没有特殊要求
3、安装vs ,此步不可缺少
4、安装cuda工具包,一定要注意版本,tensorflow.不必装最新的,2.4不支持python3.7,一般装tensorflow.2.1就可以,对应的cuda版本是10.1,我选的10.1updata 2,版本,安装时一定注意360的权限提醒,选同意,或者在安装过程中关闭360,否则会失败
5、安装cudnn,这里不必拘泥于tensorflow官网上提供的7.4版本,因为7.4没有对应的cuda10.1版本,本人选的cuda10.1对应的win10 7.64版本(cudnn-10.1-windows10-x64-v7.6.4.38),文件保存位置可百度,最简单的方法看系统变量里cuda10.1的位置,三个文件夹直接保存进去就行
6、安装tensorflow 2.1.0,提示一下清华、科大镜像此版本总掉线,阿里云无此版本,最后在豆瓣镜像成功快速安装
截至这里应该就成功安装了
备注,如果幸运,其实在第三步后可以在anaconda3的Environments里选tensorflow-gpu,anaconda3会自动生成安装计划,cuda等都不必安装,anaconda会全部代办,最为省事快捷,不过受网络影响很难成功
二、安装pytorch
安装过tensorflow.-gpu后,安装pytorch要容易的多,直接在anaconda3的Environments里选就行,因为包相对较小,很容易成功,也最省心

你可能感兴趣的:(python,tensorflow,pytorch,深度学习)