如何让代码在 Windows 10 上的 GPU 上运行？

【问题标题】：How to make code run on GPU on Windows 10?如何让代码在 Windows 10 上的 GPU 上运行？
【发布时间】：2021-08-03 20:23:57
【问题描述】：

我想在 Windows 10 的 gpu 上运行我的代码，就像 google colab 一样，我们可以更改运行时选项，这很容易转移到 gpu。是否有可能对 windows 中的 jupyter notebook 做同样的事情。

【问题讨论】：

听起来你的意思是 tensorflow？不是jupyter
@OneCricketeer 不是 tensorflow，我想将它用于 PyTorch
我明白了。无论如何，Jupyter 运行的是 Python 内核，因此您在其中编写的任何代码都应该与主机本身一样运行，因此说“在 gpu 上运行 jupyter”并不准确
@OneCricketeer 是的，正如您所说，它正在运行 python 内核，这就是为什么创建深度学习模型需要无数天的原因。出于同样的原因，我希望 jupyter 在 gpu 上运行以减少计算时间。
可以查看下面的答案stackoverflow.com/questions/51002045/…

【解决方案1】：

您实际上需要使用 tensorflow-gpu 在 gpu 上运行您的 jupyter notebook。

实现这一目标的最佳方法是

【讨论】：

嘿，我的系统上有一个 nvidia 940mx（计算能力 - 5.0），所以我应该下载旧版本的 CuDNN 和 Cuda Toolkit 还是它可以与最新版本一起使用？谢谢。
嗨，我无法导入 tensorflow-gpu，它说语法错误。虽然我可以使用“import tensorflow”，但是如何确定 jupyter 是否使用 gpu。