






















在powershell中输入:nvidia-smi

当前CUDA版本为12.6,表示支持最高版本的CUDA是12.6,可以向下兼容低版本的CUDA
NVIDIA官网: https://developer.nvidia.com/cuda-toolkit-archive
用上面的网址下载适配的版本。我用的是windows11,用11.6这个版本。


下载好之后默认安装即可
安装完成后查看系统变量:
在“系统属性”--“环境变量”中查看

在powershell输入nvcc --version验证安装

pytorch官网的老版本安装链接:https://pytorch.org/get-started/previous-versions/

我用pycharm做IDE,是用pip安装,复制对应这行命令安装。其它的开发环境对应复制官网的命令。

也可以用离线下载的方式下载安装,Pytorch下载链接:https://download.pytorch.org/whl/torch_stable.html.
在powershell中输入pip list验证

验证开发环境

若 torch.cuda.is_available() 结果为True,则表示GPU版本安装完成
此内容由惯性聚合(RSS阅读器)自动聚合整理,仅供阅读参考。 原文来自 — 版权归原作者所有。