一、基础

  • 查看版本

    1
    conda --version

  • 查看环境

    1
    conda config --show

  • 创建环境

    1
    conda create -n env_name python=3.8

  • 激活环境

    1
    conda activate env_name

    或

    1
    activate env_name

二、安装pytorch

  • 查看CUDA版本

    1
    nvidia-smi

  • cuda 12.1

    1
    conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
    1
    pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121