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


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM