TensorFlow GPU版安装方法

首先下载CUDA https://developer.nvidia.com/computeworks不推荐最新版本在这里插入图片描述
下载完成后,选择路径并安装
在这里插入图片描述
安装完成后下载https://developer.nvidia.com/cudnn对应的cudnn
将cudn解压后得到的三个文件复制到CUDA的安装文件目录下
在这里插入图片描述
至此显卡运算平台搭建完成,接下来就是Python和TensorFlow的安装
PS:Python和TensorFlow的版本是否匹配。
安装Anaconda(https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
在这里插入图片描述
完成后打开Anaconda Prompt,进入Anaconda命令行管理界面。配置清华仓库镜,输入命令:

Conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

创建运行环境,输入命令:

conda create -n tensorflow-gpu python=3.6

新建一个名字叫“tensorflow-gpu”的运行环境。
输入命令:

conda activate tensorflow-gpu 

激活并进入环境

安装tensorflow1.9,输入指令:

conda activate tensorflow-gpu

至此,tensorflow的安装完成。
如果能运行下面的代码并且能看见一些显卡的信息就表示安装成功

import tensorflow as tf
hello = tf.constant('Hello,TensorFlow!')
sess=tf.Session()
print(sess.run(hello))

End

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值