win10安装pytorch + cuda

1:下载cuda工具cuda-toolkit

地址:https://developer.nvidia.com/cuda-toolkit-archive

2:一路向下:安装

3:配置环境变量

path 环境变量中添加:

4:验证cuda是否安装成功:

nvcc -V

set cuda

5:安装cuDNN

登录cuDNN下载网页(必须注册账号并登录)并安装
cuDNN Archive | NVIDIA Developer

将解压后的文件夹中的bin、include、lib 三个文件夹,移动到CUDA Development 安装路径下,

与同名文件夹合并。

6:安装对应pytorch版本:

Start Locally | PyTorch

选择对应的版本,安装:

复制代码
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

6:验证cuda是否成功:

python 复制代码
import torch
print(torch.cuda.is_available())
相关推荐
weiwei228442 分钟前
神经网络模型导出及开放标准格式ONNX
pytorch·onnx
新新技术迷22 分钟前
Node给AI接口做SSE代理与鉴权
人工智能
ServBay1 小时前
9 个 Python 第三方库推荐,不用 AI 都好像多出一个团队
后端·python
用户8356290780511 小时前
如何使用 Python 添加和管理 Excel 批注(完整示例)
后端·python
redreamSo1 小时前
大模型是不是到顶了?瓶颈到底在哪
人工智能·openai
用户8356290780511 小时前
使用 Python 管理 Excel 工作表:创建、复制、删除与重命名
后端·python
Oo9201 小时前
Tool Use 背后的技术逻辑
人工智能
姗姗来迟了1 小时前
Vue3封装AI流式对话组件踩坑实录
人工智能
码上天下2 小时前
用Pinia管理AI多会话状态
人工智能
用户054324329702 小时前
Next.js接大模型流式SSE实操踩坑
人工智能