windows安装CUDA

本文介绍了在Windows系统上安装CUDA和PyTorch的方法。首先说明适用于英伟达显卡,接着通过查看PyTorch官网确定支持的CUDA版本,下载对应版本的CUDA并安装,测试安装是否成功,最后安装带CUDA版本的PyTorch。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

windows安装CUDA

说明

  • 适用于你的电脑为nvidia的显卡

第一步

    1. 先运行nvcc -V 查看cuda版本,不过应该都是未识别的命令,如下图所示
      在这里插入图片描述
    1. 打开pytorch的官网 https://pytorch.org/get-started/locally/ ,查看其支持最新的cuda为多少版本,例如:当前支持到cuda11.8

在这里插入图片描述

第二步

    1. 下载cuda:传送门https://developer.nvidia.com/cuda-toolkit-archive
    1. 选择版本11.8,如果后面pytorch支持更高的cuda,可以安装新版的cuda
      在这里插入图片描述

选择win10或者win11
在这里插入图片描述

  • 3.等待下载完成,双击cuda_11.8.0_522.06_windows.exe运行,直接下一步。。。

第三步

  • 测试是否正确安装,需要新打开一个终端,输入nvcc -V 如图
    在这里插入图片描述

最后

  • 再安装带cuda版本的pytorch: pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

  • 这一步应该可以在安装cuda之前安装,本人没有测试过,只实验了先安装cuda,再安装pytorch

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值