Windows10使用Anaconda安装TensorFlow-GPU
内容导读
互联网集市收集整理的这篇技术教程文章主要介绍了Windows10使用Anaconda安装TensorFlow-GPU,小编现在分享给大家,供广大互联网技能从业者学习和参考。文章包含1728字,纯文字阅读大概需要3分钟。
内容图文
简介
需要开发Unity AI Agent,可以使用tensorflow进行训练,而Unity只有Win和Mac版本的,所以重新回归Win10系统。。不过,win10也可以使用Linux的子系统了,以后就用win10和linux子系统吧。。。。
这里有Ubuntu16.04的教程:https://blog.csdn.net/qq_35976351/article/details/79325476
安装CUDA-9和CUDNN-7
选择local版本,直接安装。后边几个是一些优化的包,可以选择性的安装。
安装时,可以选择安装目录,但是要记住安装目录,后期需要添加环境变量。。我安装在了C盘自定义software目录。。
安装CUDNN,给出CUDNN的下载地址:https://developer.nvidia.com/rdp/cudnn-download 。这个需要注册账号
之后选择win10版本下载并解压:
解压后把解压后的目录中的3个文件夹,直接复制到CUDA的安装目录中,比如:
安装Anaconda
Anaconda下载地址:https://www.anaconda.com/download/
选择合适的版本,进行下载安装即可。
记住安装的地址,需要添加环境变量。进入到高级系统设置,然后选择环境变量,在下方的系统环境变量的path中,添加Anaconda的两个环境变量:
貌似CUDA的环境变量自动添加了。。。。。
安装Tensorflow-GPU
打开Anaconda的Prompt命令行窗口,创建Tensorflow的虚拟开发环境:
conda create --name tensorflow_gpu python=3.5
之后,激活环境:
activate tensorflow_gpu
之后安装tensorflow-gpu的组件,使用清华大学的源。。。
pip3 install --upgrade https://mirrors.tuna.tsinghua.edu.cn/tensorflow/windows/gpu/tensorflow_gpu-1.3.0rc0-cp35-cp35m-win_amd64.whl
之后使用Pycharm作为IDE,进行验证:
import tensorflow as tf
mat1 = tf.constant([[3, 3]])
mat2 = tf.constant([[2], [2]])
product = tf.matmul(mat1, mat2)
with tf.Session() as sess:
result = sess.run(product)
print(result)
运行程序,第一次加载的时间比较慢,红色的提醒是我们没有进行编译,可以忽略这些。结果如图:
内容总结
以上是互联网集市为您收集整理的Windows10使用Anaconda安装TensorFlow-GPU全部内容,希望文章能够帮你解决Windows10使用Anaconda安装TensorFlow-GPU所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错,欢迎将互联网集市网站推荐给程序员好友。
内容备注
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 gblab@vip.qq.com 举报,一经查实,本站将立刻删除。
内容手机端
扫描二维码推送至手机访问。