PyTorch 是当今非常火爆的深度学习框架。
相关链接
- 官网：https://pytorch.org/
- GitHub：https://github.com/pytorch/pytorch
- 官方文档：https://pytorch.org/docs/stable/index.html
安装方法
PyTorch 适配的平台有很多，官方也提供了安装教程，可以参考 https://pytorch.org/get-started/locally/，
打开之后可以看到有好多选项配置，让我们选择我们当前的环境是怎样的，如图所示：
比如这里我们可以选择版本、操作系统、包工具、开发语言、GPU 环境等等，选择之后，最后就会显示一条命令，比如这里的命令就是：
1 |
pip3 install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html |
这样的话我们只需要运行这条命令，那就可以在 Linux 上使用 pip 安装适配 CUDA 11.1 的 PyTorch 库了。
更详细的安装配置可以参考：https://pytorch.org/get-started/locally/。
来源：https://cuiqingcai.com/31073.html
