我将使用 pytorch 进行深度学习应用程序。我的计算机已经配备了 NVIDIA GPU。
我按照https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html安装了 CUDA
接下来，我使用 pytorch.org 上建议的 pip 命令在 conda 环境中安装了 pytorch
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
但是，当我检查我安装的 CUDA 版本时
nvcc - version，我得到
... Cuda compilation tools, release 12.0, V12.0.76 Build cuda_12.0.r12.0/compiler.31968024_0
当我运行代码时，我的安装似乎没问题
torch.cuda.is_available()
因为它返回 True 并正确识别 GPU 的名称。
只是想检查一下，当我实际输入具有大数据集的深度学习模型时，它是否万无一失，因为 CUDA 版本是 12.0，但我安装了适合 11.6 的 pytorch。
已经非常感谢了。