1、使用PyTorch进行深度学习时,为什么提到CUDA?什么是CUDA?它和PyTorch的关系是?
CUDA 是 NVIDIA 提供的用于 GPU 加速计算的平台和编程模型;PyTorch 是一个基于 Python 的科学计算库。可以将它们视为两个不同的软件,都需要单独安装,而它们的关系是:PyTorch 可以通过 CUDA 利用 GPU 加速计算。因此,如果你下载的PyTorch为CPU版本,进行深度学习就完全不需要下载CUDA;PyTorch为GPU版本时才需要再下载CUDA。我太懒了就直接选的PyTorch的CPU版本然后做实验(虽然我的笔记本也并没有独立GPU)。
2、PyTorch的下载
非常简单,进入PyTorch的官网,选择自己需要的选项,复制下面的语句粘贴到命令行运行即可。
3、CUDA的安装
进入官网,选择自己需要的选项,点击下载即可,然后安装下载好的.exe文件。
4、PyTorch的版本和CUDA的版本必须要匹配哦。