Windows10安装TensorFlow GPU版本

    xiaoxiao2021-03-25  134

    本文描述的是如何在windows 10 64位操作系统安装TensorFlow,采用预编译版本。

    前提条件

    一块支持CUDA 8.0的nvidia显卡。

    可以在官方网站查询显卡的兼容性信息。

    安装nvidia cuda 8.0

    https://developer.nvidia.com/cuda-downloads

    从官方网站下载安装包,一路next就可以了。

    确保cuda的路径信息加入PATH,不然运行会找不到nvcuda.dll。

    安装cuDNN

    https://developer.nvidia.com/cudnn

    下载安装包后解压缩,并把cudnn64_5.dll的路径加入PATH。

    安装Anaconda

    从https://www.continuum.io/downloads下载软件并安装。

    安装TensorFlow

    conda create --name tensorflow python=3.5

    activate tensorflow

    pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.1.0-cp35-cp35m-win_amd64.whl

    验证安装

    #coding=utf-8 #单机单卡 #对于单机单卡,可以把参数和计算都定义在gpu上,不过如果参数模型比较大,显存不足等情况,就得放在cpu上 import tensorflow as tf with tf.device('/cpu:0'):#也可以放在gpu上 w=tf.get_variable('w',(2,2),tf.float32,initializer=tf.constant_initializer(2)) b=tf.get_variable('b',(2,2),tf.float32,initializer=tf.constant_initializer(5)) with tf.device('/gpu:0'): addwb=w+b mutwb=w*b ini=tf.initialize_all_variables() with tf.Session() as sess: sess.run(ini) np1,np2=sess.run([addwb,mutwb]) print(np1) print(np2)

    --结束--

    转载请注明原文地址: https://ju.6miu.com/read-3473.html

    最新回复(0)