在GPU上设置机器学习和深度学习感到困惑

我的目标是通过GPU为我的PC设置机器学习和深度学习环境。我已经阅读了关于各种组件的信息,但是我无法将这些信息串联起来,搞清楚我需要做什么。

  • 操作系统:Ubuntu 20.04
  • GPU:Nvidia RTX 2070 Super
  • Anaconda:4.8.3

我已经安装了nvidia-cuda-toolkit(10.1.243),但接下来该做什么呢?

  • 这如何与Jupyter Notebook集成?
  • 我想使用的3个Python模块是:
    • turicreate – 我已经在CPU上运行了它,但还没有在GPU上运行
    • scikit-learn
    • tensorflow
    • matlab

我知道cuDNN和pyCUDA应该在某个地方起作用。

任何帮助都将不胜感激。谢谢


回答:

首先,我的经验仅限于Ubuntu 18.04和16.xx以及Python深度学习框架。但我希望一些建议会对你有帮助。

  • 如果我熟悉Docker,我会考虑使用Docker而不是从头开始设置所有东西。这种方法在关于TensorFlow容器的部分中有描述
  • 如果你决定自己设置所有组件,请参考这个指南,我曾用其中的一些内容在18.04上成功设置
  • 要小心自动更新。配置完成并测试后,保护它不被CUDA或TensorRT的最新版本覆盖。

回答你其中一个子问题 – 这如何与Jupyter Notebook集成?它不需要集成,因为这是不必要的。CUDA库与像TensorFlow这样的框架合作,而不是与Jupyter。Jupyter只是服务器端的一个编辑器和执行控制器。

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注