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版本

 

转载请注明出处:  https://www.cntworld.cn
智能工控 » cuda版本升级与Torch安装

发表回复

提供最优质的资源集合

立即查看 了解详情