如何在Anaconda中安装PyTorch和TensorFlow？

安装PyTorch和TensorFlow的步骤

安装PyTorch
打开Anaconda Prompt或终端，创建一个新的conda环境（可选但推荐）：
conda create -n pytorch_env python=3.8
激活环境：
conda activate pytorch_env

通过PyTorch官网推荐命令安装（以CUDA 11.3为例）：
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
若无GPU支持，使用CPU版本：
conda install pytorch torchvision torchaudio cpuonly -c pytorch

安装TensorFlow
创建独立conda环境（可选）：
conda create -n tf_env python=3.8
激活环境：
conda activate tf_env

安装TensorFlow（GPU版本需提前配置CUDA/cuDNN）：
pip install tensorflow
对于GPU版本：
pip install tensorflow-gpu

验证安装
PyTorch验证：

import torch
print(torch.__version__)
print(torch.cuda.is_available())  # 检查GPU支持

TensorFlow验证：

import tensorflow as tf
print(tf.__version__)
print(tf.test.is_gpu_available())  # 检查GPU支持

注意事项

• 确保conda或pip版本已更新：conda update conda / pip install --upgrade pip
• PyTorch和TensorFlow的GPU版本需提前安装对应版本的NVIDIA驱动、CUDA和cuDNN。
• 若出现冲突，优先使用虚拟环境隔离不同框架的依赖。