[英]Tensorflow on visual studio code ( or jupyter notebook) with gpu acceleration
如何在 Visual Studio 代码(Windows 11)上激活 GPU 加速以使用 tensorflow 计算神经网络? 显卡 = 英伟达 gtx 1070 ti
将解压出来的CUDA文件夹添加到系统环境变量中(到bin目录下)
使用pip install tensorflow
安装 tensorflow 包
测试:
导入张量流 导入张量流作为 tf
打印(tf.test.is_gpu_available())
如果你得到True ,则意味着成功。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.