pytorch安装 CUDA安装 pytorch开发环境配置win10

pytorch开发环境配置(win 10)

建议先更新NVIDIA显卡驱动。

  1. 首先查看系统的显卡驱动所支持的CUDA版本:右击桌面右下角小图标[NVIDIA设置] -> 左击[NVIDIA控制面板] -> 点击左下角[系统信息] -> 选择[组件]栏,将看到下图,第三行中的[产品名称]为CUDA 11.1.70,这是你的显卡驱动所支持的最高版本的CUDA
    NVIDIA组件-所支持CUDA最高版本为11.1.70
  2. 前往NVIDIA官网下载对应版本)的CUDA并安装。注意,pytorch 1.6最高支持CUDA 10.2,所以最好下载CUDA 10.2版本。安装时选择自定义安装,取消多余组件的安装:自定义安装-取消无用组件自定义安装-取消无用组件
    安装CUDA后,运行cmd,输入nvcc --version,若显示CUDA的版本号,则表示CUDA安装成功。
  3. 开始本步前,请先自行安装python3.6环境。前往pytorch官网,可看到下图,按照你自己的系统进行选择,CUDA版本号一定要选前面步骤安装的CUDA版本。最下面的框里自动生成安装pytorch的命令,复制下来粘贴到cmd执行即可成功安装pytorch(支持gpu加速)pytorch官网首页-安装pytorch
  4. 进入python交互测试pytorch是否成功,直接在cmd输入python即可进入。然后执行
import torch
torch.cuda.is_available()

若输出True,则表示pytorch可在GPU上运行。

©️2020 CSDN 皮肤主题: 书香水墨 设计师:CSDN官方博客 返回首页