Ubuntu系统快速安装PyTorch

1. 下载conda
wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-2021.11-Linux-x86_64.sh
2. 安装conda
bash Anaconda3-2021.11-Linux-x86_64.sh并按提示同意安装协议

3. 启动环境变量
source ~/.bashrc
4. 安装PyTorch
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

    所属分类：Python     发表于2022-08-02