博主原文链接:https://www.yourmetaverse.cn/nlp/466/
Anaconda是数据科学和机器学习领域中广泛使用的Python分发版。它简化了包管理和部署过程,特别是在处理复杂的依赖关系时。在本教程中,我将指导你如何在Anaconda中创建独立的PyTorch和TensorFlow环境,并在每个环境中安装相应的Jupyter Notebook kernel。
首先,确保你的系统中安装了Anaconda。你可以从Anaconda官网下载并遵循安装指南进行安装。
打开Anaconda命令行界面。
pytorch_env
:conda create -n pytorch_env python=3.8
conda activate pytorch_env
tensorflow_env
:conda create -n tensorflow_env python=3.8
conda activate tensorflow_env
确保你在正确的环境中。
在pytorch_env
环境中,运行以下命令:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
根据你的系统配置选择合适的CUDA版本。
在tensorflow_env
环境中,运行以下命令:
conda install tensorflow
pytorch_env
环境中。conda install ipykernel
python -m ipykernel install --user --name=pytorch_env --display-name="PyTorch Kernel"
tensorflow_env
环境中。conda install ipykernel
python -m ipykernel install --user --name=tensorflow_env --display-name="TensorFlow Kernel"