我的目标是通过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只是服务器端的一个编辑器和执行控制器。