神经网络与深度学习-2

一、深度学习平台

PyTorch是一个开源的Python机器学习库，基于Torch，用于自然语言处理等应用程序。

安装Pytorch

PyTorch的安装十分简单，根据PyTorch官网，对系统选择和安装方式等灵活选择即可。这里以anaconda为例。Pytorch的安装经过了几次变化，请大家以官网的安装命令为准。另外需要说明的就是在1.2版本以后，Pytorch只支持cuda 9.2以上了，所以需要对cuda进行升级，部分显卡都可以用，包括笔记本的MX250也是可以顺利升级到cuda 10.1。此处使用Conda包管理器。注意：如果使用镜像站，请删除“-c pytorch”；安装CUDA（即GPU）版本时注意安装CUDNN运行库。

# 全部通用，ROCm仅支持Linux
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch    #CUDA 10.2
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch    #CUDA 11.3
conda install pytorch torchvision torchaudio cpuonly -c pytorch