Win10 在 CUDA 10.1 下跑 TensorFlow 2.x
2024-09-03 12:30:43
深度学习最热的两个框架是 pytorch 和 tensorflow,pytorch 最新版本是 1.3,tensorflow 最新版本为 2.0,在 win10 下 pytorch 1.3 要求的 cuda 最高版本是 10.1,见下图:
而 tensorflow 2.0 使用的 cuda 版本是 10.0,见下图:
这就造成了冲突,一般是装 cuda 10.1,然后再重新编译 tensorflow 2.0 源码跑在 cuda 10.1 下,编译 tensorflow 源码的步骤还是较麻烦的,也很容易失败,不过现在有更好的方案了,
就是直接安装 NIGHTLY 版本,目前最新版本为 tensorflow 2.1.0-rc2,使用 pip 安装
pip install tensorflow-gpu==2.1.0-rc2
检验一下
import tensorflow as tf
print(tf.test.is_gpu_available())
发现已经能正常使用 gpu 了
2020-01-01 10:40:26.616452: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1241] Created TensorFlow device (/device:GPU:0 with 4702 MB memory) -> physical GPU (device: 0, name: GeForce GTX 1060, pci bus id: 0000:01:00.0, compute capability: 6.1)
True
最新文章
- java 对象序列化与反序列化
- java并发编程资料
- CodeForces 520B Two Buttons(用BFS)
- Sprint 2
- ios第二天{函数}
- js-PC版监听键盘大小写事件
- 轻松自动化---selenium-webdriver(python) (五)
- 全程图解 手把手教您开启windows终端服务
- HTML-css selector
- jQuery插件实现select下拉框左右选择_交换内容(multiselect2side)
- <;转>;golang 并发性能数据
- apache2.4.4启用deflate压缩
- 《Java从0开始的成长之路》
- 《程序设计方法》【PDF】下载
- 探寻 webpack 插件机制
- HAProxy负载均衡技术
- cdnbest站点里设置防盗链
- iptable用法
- 25. Spring Boot与缓存 JSR-107、Spring缓存抽象
- CentOS6.5下实现R绘图
热门文章
- 对象析构谈—— delete this 的使用及注意事项
- 虚拟机kali Linux 的网络配置
- 【Webex】加入会议是无法正常加入!提示“下载会议组件时无法获取正确的参数。 请联系技术支持以获取帮助。
- maven中的pom.xml中的scope的作用
- PHP基础学习笔记2
- 算法刷题--回溯算法与N皇后
- JavaScript可枚举的属性
- 【PAT甲级】1070 Mooncake (25 分)(贪心水中水)
- 【实战】Springboot +jjwt+注解实现需登录才能操作
- SQL Server 2014数据库开启远程连接(Windows Server 2016)