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

Posted 2023-03-12

【中文标题】如何让代码在 Windows 10 上的 GPU 上运行？

【英文标题】：How to make code run on GPU on Windows 10?

【发布时间】：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 上运行以减少计算时间。

可以查看下面的答案***.com/questions/51002045/…

【参考方案1】：

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

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

在您的系统上安装Anaconda

下载cuDNN和Cuda Toolkit 11.3。

将 cuDNN 和 Cuda Toolkit 添加到您的 PATH。

在 Anaconda 中创建环境

pip install tensorflow-gpu

pip install [jupyter-notebook/jupyterlab]

在你的 notebook 中导入 tensorflow-gpu

享受。您现在可以在 GPU 上运行笔记本

【讨论】：

嘿，我的系统上有一个 nvidia 940mx（计算能力 - 5.0），所以我应该下载旧版本的 CuDNN 和 Cuda Toolkit 还是它可以与最新版本一起使用？谢谢。

嗨，我无法导入 tensorflow-gpu，它说语法错误。虽然我可以使用“import tensorflow”，但是如何确定 jupyter 是否使用 gpu。