date
Nov 1, 2025
slug
2025-11-01-the-installation-process-of-CUDA-pytorch-developmemnt-environment
status
Published
tags
LLM
summary
本文基于Windows 11操作系统和Nvidia RTX5070Ti显卡硬件完成了Cuda+Pytorch开发环境的搭建与测试验证。
type
Post
category
AI
AI summary
操作系统:Windows 11
硬件:HP暗夜精灵Max,Ultra9 275HX+5070Ti
检查Nvidia显卡驱动版本
在终端输入nvidia-smi:
可以看到以上显卡驱动支持的CUDA版本最高是12.9。
- 如果以上命令提示找不到,就需重新安装Nvidia的显卡驱动:Download The Official NVIDIA Drivers | NVIDIA
安装Pytorch的GPU版本
在Pytorch官方页面Get Started根据当前的CUDA版本选择安装对应的torch版本:因为前面查到的当前CUDA版本是12.9,所以这里选择的版本应该是CUDA12.8的Pytorch版本,复制下面的安装命令安装pytorch library。
CUDA版本的匹配规则:PyTorch的CUDA版本应≤驱动支持的版本

测试验证Pytorch的环境
以上安装完成后使用下面的脚本测试当前环境是否已经安装好:
正确的情况下,上面脚本执行后的打印消息为: