本文介绍了如何在不同环境下安装PyTorch，包括检查CUDA和cuDNN版本，以及针对无CUDA环境的Mac系统使用pip安装的方法。同时提供了加速pip安装的镜像源设置指南，确保顺利安装PyTorch、torchvision和torchaudio库。
Python——PyTorch的安装
越来越多的人选择PyTorch的深度学习框架了，那么这么包如何安装呢？注意，它不能像Tensorflow或者Keras那样安装了，否者会报错的。官网给了具体说明，咱们移步：PyTorch官网
注意：
- 查看cuda版本：
-
方法1: cat /usr/local/cuda/version.txt 方法2: nvcc --version
- 查看cudnn版本：
-
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
比如，我的环境是Mac，无CUDA，pip安装，那就直接是：
pip install torch torchvision torchaudio如果安装比较慢，配置一下镜像源，参考以前的笔记：Mac——三步更改Python 默认pip源
验证一下：
扫一扫
7956
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
