为什么有GPU,使用pytorch训练的时候还是使用的CPU训练?

为什么有GPU,使用pytorch训练的时候还是使用的CPU训练?

  • 前言
    • `提示:因为使用pip进行安装,默认下载的就是cpu的版本:` ![在这里插入图片描述](https://img-blog.csdnimg.cn/1c4dc1d99bb84b1596685e4cd9b7497e.png)
  • 解决方法
    • 1、进入https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/下载安装包
    • 2、选择对应的cuda版本进行下载
      • 如果不知道cuda的版本在控制台中输入nvidia-smi查看
    • 3、下载对应的pytorch在本地进行安装就行了


前言

提示:因为使用pip进行安装,默认下载的就是cpu的版本:
在这里插入图片描述

解决方法

1、进入https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/下载安装包

为什么有GPU,使用pytorch训练的时候还是使用的CPU训练?_第1张图片

2、选择对应的cuda版本进行下载

如果不知道cuda的版本在控制台中输入nvidia-smi查看

为什么有GPU,使用pytorch训练的时候还是使用的CPU训练?_第2张图片

3、下载对应的pytorch在本地进行安装就行了

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