Tensorflow-gpu安装记录
安装配置
- 系统:Windows10
- 显卡:QTX1080Ti
- ViisualStudio:2017
- Python:3.7.3
- CUDA:10.0
- cuDNN:7.6.0
- Tensorflow-gpu:1.13.1
查看当前显卡信息
- 在桌面空白处单击右键进入NVIDIA控制面板
- 单击界面左下角的系统信息
- 可在显示和组件两个选项卡中查看显卡信息
- GPU,CUDA,cuDNN的理解
- 驱动与CUDA的关系
我的显卡驱动版本是411.31,所以我装的是CUDA10.0
- CUDA支持的GPU
页面上给出了支持CUDA的GPU型号和对应的算力
CUDA支持GTX1080Ti,且这款GPU的算力为6.1
安装Visual Studio2017
- 安装CUDA需要提前安装好Visual Studio,进入官网下载VS2017,社区版是免费的
- 勾选通用Windows平台开发、使用C++的桌面开发、Python开发、Visual Studio 扩展开发
配置Python环境
安装Python
进入Python官网下载Python3.7.3,安装过程中记得勾选加入环境变量那一项,这样就不用手动添加了
也可直接下载Anaconda搭载Python3.7版本的软件,下载地址:这里
安装virtualenv库
win+R 输入cmd进入命令行,在命令行输入
pip install virtualenv
创建虚拟环境
cd 进入希望存放虚拟环境的地址,这里方便演示,直接在F盘下创建了一个虚拟环境
virtualenv test_env
test_env可替换为你自己取的虚拟环境的名字
创建完成后可以看到创建的虚拟环境文件夹