PyTorch安装及配置

PyTorch 环境配置及安装

1. 安装 anaconda

2. 为 PyTorch 创建单独的环境,便于管理。

  1. 打开 Anaconda Prompt

  2. 为 PyTorch 创建单独的环境
    python 版本与 anaconda 中 python 版本相同即可。
    conda create -n pytorch python=3.8

    输入 y 开始安装需要的包并创建环境(如果出现了 ERROR 并且你此时正在使用代理,尝试关闭代理)。
    创建成功
    创建成功。

  3. 激活环境并查看环境下的包
    conda activate pytorch
    pip list

  4. 如果没有 pytorch 包,需要下载安装。

3. PyTorch 安装

  1. pytorch 首页:https://pytorch.org/

    选择相关参数,如果电脑没有Nvidia显卡或者只有集显,CUDA 选择 CPU 即可。然后运行下面的指令。

  2. 在 pytorch 环境下运行指令。

  3. 安装检验
    命令行输入 python 进入 python 环境,执行 import torch ,如果没有报错,说明安装成功。
    输入 torch.cuda.is_available() 检查 GPU 是否可用, 返回 true 即表示可用。

至此,PyTorch 的安装配置完成。


4. 在新建的 pytorch 环境中安装 Jupyter

  1. 进入 pytorch 环境 conda activate pytorch

  2. 安装需要的包 conda install nb_conda

  3. 屏幕上显示 done 即安装成功。

  4. 使用 Jupyter jupyter notebook

posted @ 2021-11-17 11:11  少年与梦3512  阅读(639)  评论(0)    收藏  举报