cuda版本升级与Torch安装
当电脑运行大程序序时经常报错才发现cuda版本只有11.8,而且用nvidia-smi查出来是支持的最高版本12.3版本,而不是当前的11.8版本;
查看现有系统版本nvcc -V,而且没有Build cuda_12.3.r12.3/compiler…字样;
去官网下载cuda版本,根据自己的需求下载安装,本机是windows 11操作系统;
卸载旧版本CUDA,在控制面板中卸载旧版;
安装下载的CUDA软件,安装时关闭所有的游戏、防毒软件、图形设计软件等与GPU相关的软件。再次检测版本信息就发现已经安装了cuda_12.3了;
如果是conda虚拟环境的话通过pip show torch命令查看PyTorch包的安装信息
PyTorch软件公司的算法平台torch下载,以管理员权限运行PowerShell命令;
# CUDA 12.1
conda install pytorch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 pytorch-cuda=12.1 -c pytorch -c nvidia
在PowerShell中运行pip show torch查看当前torch的版本;
检查环境变量的配置是否为已安装的cuda_12.3版本