pytorch指定cuda

CUDA是英伟达公司推出的一种并行计算平台和编程模型，可以利用GPU的强大计算能力加速计算任务。

cuDNN是CUDA深度神经网络库，提供了高效的卷积神经网络和循环神经网络的实现，可以加速深度学习模型的训练和推理。

PyTorch是一个基于Python的科学计算库，它提供了强大的GPU加速功能，支持动态图和静态图两种计算图模式，是深度学习领域中广泛使用的框架之一。