一起动手学习神经网络——Pytorch和PYG包的安装
看了多篇安装博客的内容,本想应该很简单,没想到却是费了一番功夫,现把自己安装的过程写下来,以便在后续过程中其他需要小伙伴能少走一些弯路。
第一步:安装Visual Studio 2017.
要勾选上【 C++/CLI support】和【VC++ 2015.3 v14.00(v140) toolset for Desktop】两个在visual C+++ build tools下面的包;
先下载在线安装版,运行起来后选择性安装。
除上面必要的,当然也可以添加其他的组件
等待安装完成。
注意好多博客里都没有提到这一步,但如果不先进行第一步,直接安装cuda会在最后一步报没有Visual Studio的错误,点击忽略安装,看似成功,其实并没有。
第二步:安装cuda和 cuDNN。
先简单了解一下:
- 什么是 CUDA
CUDA(ComputeUnified Device Architecture),是显卡厂商NVIDIA推出的运算平台。CUDA是一种由NVIDIA推出的通用并行计算架构,该架构能使GPU能够解决复杂的计算问题。
- 什么是cuDNN
NVIDIA cuDNN是用于深度神经网络的GPU加速库。它强调性能、易用性和低内存开销。
NVIDIA cuDNN可以集成到更高级别的机器学习框架中,如谷歌的Tensorflow、加州大学伯克利分校的流行caffe软件。简单的插入式设计可以让开发人员专注于设计和实现神经网络模型,而不是简单调整性能,同时还可以在GPU上实现高性能现代并行计算。
- CUDA 和 cuDNN 的关系
CUDA看作是一个工作台,上面配有很多工具ÿ