1.CUDA是什么?
CUDA(Compute Unified Device Architecture),显卡厂商NVidia推出的运算平台。
随着显卡的发展,GPU越来越强大,而且GPU为显示图像做了优化。在计算上已经超越了通用的CPU。如此强大的芯片如果只是作为显卡就太浪费了,因此N卡厂商推出CUDA,让显卡可以用于图像计算以外的目的。
目前只有G80平台的N卡才能使用CUDA,工具集的核心是一个C语言编译器。G80中拥有128个单独的ALU,因此非常适合并行计算,而且数值计算的速度远远优于CPU。
CUDA的SDK中的编译器和开发平台支持Windows、Linux系统,可以与Visual Studio2003集成在一起。
2.CUDA下载地址
https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
选择操作系统的版本我的是win10
3.CUDA安装
安装文件cuda_9.1.85_win10.exe
安装前要关闭360安全卫士等类似杀毒软件等。
4.查看版本
在CMD里运行:nvcc -V
5.安装Anaconda
6.用Anaconda安装TensorFlow
在Anaconda里建立TensorFlow的环境
conda create -n tensorflow Python=3.5
激活TensorFlow环境:
activate TensorFlow
关闭环境:
deactivate
安装TensorFlow:
pip install --upgrade --ignore-installed tensorflow-gpu
安装TensorFlow指定版本(清华源上有的,更换链接最后的版本名称就行了)
pip install --upgrade https://mirrors.tuna.tsinghua.edu.cn/tensorflow/windows/gpu/tensorflow_gpu-1.3.0rc0-cp35-cp35m-win_amd64.whl
简单测试:
编写showTFversion.py文件,内容为:
import tensorflow as tf hello = tf.constant("Hello!TensorFlow") sess = tf.Session() print(sess.run(hello))
激活TensorFlow环境:
activate TensorFlow
在TensorFlow环境里运行showTFversion.py文件: