TensorFlow-gpu环境配置

1、安装anaconda

2、新建TensorFlow-gpu环境

3、安装cuda

(1)查看gpu对应cuda版本:桌面右键->NVIDIA控制面板->帮助->系统信息->组件->查看cuda信息(图1);

(2)下载cuda并安装:cuda下载地址;(目录:C:\Program Files\NVIDIA GPU Computing Toolkit)

(3)下载cudnn:cudnn下载地址(需要账号密码登录,qq邮箱、账号annatu1206、密码含字母数字特殊字符);

           下载完成后不需要安装,解压后替换掉cuda9.0根目录下的文件就行

4、进入TensorFlow-gpu环境,安装tensorflow-gpu

(1)升级pip工具防止安装到一半报错:python -m pip install --upgrade pip

(2)安装tensorflow-gpu:pip install --ignore-installed --upgrade tensorflow-gpu

 

遇到的问题:

1、安装tensorflow-gpu有点问题,import时报错

原因:安装的cuda10.1

解决方案:卸载cuda10.1,需保留驱动、系统软件、vs相关软件;然后重新安装cuda10.0

图1:

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章