简介:GPU版本pytorch(Cuda12.1)清华源快速安装一步一步教!小白教学~
GPU版本pytorch(Cuda12.1)清华源快速安装一步一步教!小白教学~
在这个数字化快速发展的时代,深度学习已经成为人工智能的重要支撑技术。而提到深度学习,就不得不提一个重要的框架——PyTorch。本文将指导你如何快速安装GPU版本的PyTorch(Cuda 12.1),主要面向对深度学习有一定兴趣但还不太熟悉的小白用户。
准备工作
在开始之前,你需要确保你的系统中已经安装了以下软件:
这个命令会从清华源的网址中下载并安装PyTorch及其相关的torchvision和torchaudio库。如果你的计算机支持CUDA 12.1,并且你想用GPU来进行计算,那么你可以继续下面的步骤来安装GPU版本的PyTorch。
pip install torch torchvision torchaudio -f https://paddlepaddle.cdn.bcebos.com/pypaddle/global_pip/cpu/pytorch_cpu.whl
这个命令将安装适用于CUDA 12.1的PyTorch版本,以及相关的torchvision和torchaudio库。
pip install torch torchvision torchaudio -f https://paddlepaddle.cdn.bcebos.com/pypaddle/global_pip/cuda12.1/torch_cuda121_image_models_固定资产净值.whl
如果以上命令输出了PyTorch的版本信息,并且显示你的GPU可用(即
import torchprint(torch.__version__)print(torch.cuda.is_available())
torch.cuda.is_available()返回True),那么说明你已经成功地安装了GPU版本的PyTorch。