Pytorch分两个版本:GPU版本和CPU版本。
推荐安装教程:1. Pytorch零失败安装使用|Anaconda安装|最新版本torch安装|PyCharm配置torch| 保姆级教程【PyTorch修炼】【AI蜗牛车】_哔哩哔哩_bilibili
推荐博客:
Anaconda下载以及初始环境搭建pytorch、pycharm、Jupyter Notebook_conda中科大源下载pytorch_^AA^的博客-CSDN博客
1.anaconda安装
https://repo.anaconda.com/archive/Anaconda3-2023.03-Windows-x86_64.exe
2.通过conda创建一个名为:pytorch的虚拟“房间
步骤1:运行Anaconda prompt
步骤2:创建一个名为:pytorch的虚拟“房间
conda create -n pytorch python=3.7
步骤3: 激活pytorch房间
conda activate pytorch
步骤4:配置国内源
//清华源:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --set show_channel_urls true
//中科大源:
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/menpo/
conda config --set show_channel_urls yes
3.查看显卡适配器
如果只有一个显卡适配器则是集成显卡只能安装cpu版本。
4.pip安装pytorch
选择版本CPU或GPU
cpu版本:
pip3 install torch torchvision torchaudio
GPU版本:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117
安装成功:
python绝大多数源码包: