集显如何安装pytorch (CPU)

目录

  • 前言
  • 1. 注意事项
  • 2. CPU版

前言

此为记录贴,原先都是独显,安装GPU,第一次查看发现是集显,没有配备的GPU

1. 注意事项

  1. 验证自身是否为独显:cmd下执行 DxDiag

如果Graphics则为集显,代表只能操作CPU模块的
集显如何安装pytorch (CPU)_第1张图片

  1. 如果为独显,执行nvidia-smi,有这种情况:'nvidia-smi' 不是内部或外部命令,也不是可运行的程序 或批处理文件。
    一般都是没有配备环境变量

集显如何安装pytorch (CPU)_第2张图片
默认位置在C:\Program Files\NVIDIA Corporation\NVSMI,确认该位置之后,将此路径添加到系统的环境变量path中

如果满足独显又找不到配置的nvsmi执行文件,需要下载在配置:NVIDIA驱动程序下载

2. CPU版

确定自身为集显,想要安装CPU版本,请继续往下:

  • 非虚拟环境下:pip install torch torchvision torchaudio

  • 虚拟环境下:conda install pytorch torchvision torchaudio cpuonly -c pytorch

集显如何安装pytorch (CPU)_第3张图片

最终验证是否成功安装:

import torch
print(torch.__version__)

截图如下:

集显如何安装pytorch (CPU)_第4张图片

你可能感兴趣的:(深度学习,pytorch,人工智能,python)