最新消息:雨落星辰是一个专注网站SEO优化、网站SEO诊断、搜索引擎研究、网络营销推广、网站策划运营及站长类的自媒体原创博客

windows cuda更新教程

运维笔记admin3浏览0评论

目录

  • 查看 CUDA 驱动版本
  • 卸载旧版 CUDA
  • 安装 CUDA
  • 安装完成

查看 CUDA 驱动版本

查看当前安装的 CUDA 版本,当前 CUDA 版本为 10.1

nvcc -V

查看显卡驱动,以及支持的 CUDA 的最高版本,这里的12.0 指的是可驱动的最高版本。下载 CUDA 的版本应低于12.0

nvidia-smi

卸载旧版 CUDA

控制面板中卸载 CUDA

安装 CUDA

在 CUDA 官网 选择符合要求的版本安装
由于我安装的 pytorch 对应的 CUDA 版本是 11.7,所以我选择 11.7.1 版本

import torch
torch.cuda.is_available() # 查看pytorch是否支持CUDA
torch.cuda.device_count() # 查看可用的CUDA数量
torch.version.cuda # 查看对应CUDA的版本号

根据自己的系统选择并下载安装包

双击安装包,开始安装

选择自定义安装

我只卸载了CUDA,勾选CUDA

之后按默认设置,完成安装

安装完成

发布评论

评论列表(0)

  1. 暂无评论