CUDA工具包:
选择cuda版本时,注意先在NVIDIA中看电脑显卡支持的CUDA版本。
完整的 CUDA 开发工具集,包含编译器、库、驱动等,提供 CUDA 开发环境(编译 CUDA 代码、运行 CUDA 程序)。
1
2
3
4
5
6
C:\Users\name>nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Wed_Feb__8_05:53:42_Coordinated_Universal_Time_2023
Cuda compilation tools, release 12.1, V12.1.66
Build cuda_12.1.r12.1/compiler.32415258_0
声明:系统 CUDA 版本可以高于 PyTorch 编译版本,且在多数情况下推荐这么做(尤其是新显卡),只要保证系统 CUDA 版本 ≥ PyTorch 要求的最低版本即可。这不会影响项目的正常运行,还可能带来更好的硬件适配性。
Pytorch相关包:
1
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
上述本质为预编译的 PyTorch 及其依赖库,仅包含运行时组件。有预编译的 torch._C
等二进制文件(依赖 CUDA 库),核心作用在于让 PyTorch 能调用 GPU 进行计算(依赖 CUDA 运行时)。