以下是PyTorch和TensorFlow的官方推荐安装命令,适用于常见平台(Windows/Linux/macOS)和不同硬件配置(CPU/GPU)。请根据实际环境选择对应版本:
PyTorch安装命令
通过官方安装向导生成具体命令(PyTorch官网):
稳定版(CPU版本)
pip install torch torchvision torchaudio
CUDA GPU版本(以CUDA 11.8为例)
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
MacOS(Metal GPU加速)
pip install torch torchvision torchaudio
TensorFlow安装命令
稳定版(CPU版本)
pip install tensorflow
CUDA GPU版本(需提前安装CUDA/cuDNN)
pip install tensorflow[and-cuda]
旧版本指定(如2.12.0)
pip install tensorflow==2.12.0
注意事项
- GPU版本需提前安装对应版本的NVIDIA驱动、CUDA和cuDNN。
- 推荐使用Python 3.8-3.10版本,避免兼容性问题。
- 可通过
conda create -n env_name python=3.8创建虚拟环境隔离依赖。
1008

被折叠的 条评论
为什么被折叠?



