打开官网根据自己的操作系统选择对应的安装命令即可。 https://pytorch.org/get-started/locally/PyTorch 的GPU加速版是和特定CUDA版本绑定的。 命令行中输入nvidia-smi在CUDA Version栏查看显卡支持的最高CUDA版本。conda create-n nlp python3.11conda activate nlppip install torch2.0.1cu117 torchvision0.15.2cu117 torchaudio2.0.2--index-url https://download.pytorch.org/whl/cu117importtorchprint(torch.__version__)# 应显示 2.0.1print(torch.cuda.is_available())# 应返回 Trueprint(torch.version.cuda)# 应显示 11.7 03_cuda 11.7 / python3.11安装Pytorch
打开官网根据自己的操作系统选择对应的安装命令即可。 https://pytorch.org/get-started/locally/PyTorch 的GPU加速版是和特定CUDA版本绑定的。 命令行中输入nvidia-smi在CUDA Version栏查看显卡支持的最高CUDA版本。conda create-n nlp python3.11conda activate nlppip install torch2.0.1cu117 torchvision0.15.2cu117 torchaudio2.0.2--index-url https://download.pytorch.org/whl/cu117importtorchprint(torch.__version__)# 应显示 2.0.1print(torch.cuda.is_available())# 应返回 Trueprint(torch.version.cuda)# 应显示 11.7