1.更新最新的显卡驱动
cuda版本要求的最低显卡驱动版本
2.版本对应
实测:CUDA9.0-TF1.9.0,CUDA9.2-TF1.8.0(主要是云盘只有此编译包)
3.安装cuda之前首先安装vs2015,只勾选C++,其余默认,网盘4G,本notebook安装2H左右
严长生-VS2015安装教程
4.下载安装Cuda9.2及补丁,官网直接下载安装,很快,1.5G
Cuda官网地址
5.下载安装对应Cudnn版本,官网注册微信登录下载,直接解压-复制-粘贴覆盖cuda9.2文件夹,会自动写入,不用担心文件异常
cudnn官网地址
6.在Aconda prompt里创建tf-gpu环境,并激活,更新pip,可以默认pip的国内源,官网的tf安装包目前并不支持(20200504),后附上小伙伴们整理的云盘资料
conda create -n tensorflow-gpu python=3.6
conda activate tensorflow-gpu
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
python -m pip install --user--upgrade pip
http://pypi.douban.com/ # 豆瓣
http://pypi.hustunique.com/ # 华中理工大学
http://pypi.sdutlinux.org/ # 山东理工大学
http://pypi.mirrors.ustc.edu.cn/ # 中国科学技术大学
http://mirrors.aliyun.com/pypi/simple/ # 阿里云
https://pypi.tuna.tsinghua.edu.cn/simple/ # 清华大学
pip install tensorflow-gpu==1.9.0#实测不行,必须用GIThub上大神们编译过的安装包
pip install tensorflow_gpu-1.8.0-cp36-cp36m-win_amd64.whl#安装包所在文件目录下pip
7.Pycharm打开py文件,设置tf-gpu环境,里面库很少,需要自己安装,tf-gpu环境在Aconda3路径下,应用即可
参考网址:
1-Jaykie_WIN10+tensorflow1.9.0-GPU+CUDA9.0+vs2015+python3.6安装教程
2-sweet virus-RTX2060+win10+Tensorflow GPU+CUDA 9.2+CUDNN7.2+python 3.65
3.感谢cduter小伙伴,不会翻墙,用了小伙伴提供的tf1.8.0安装包
资源链接: https://pan.baidu.com/s/1UTBCijit2JVEP11ge7XLCg 密码: hlol