一、安装 CUDA
1、安装CUDA前请先确认Torch 所支持的版本,以免安装的Torch 不支持最新CUDA的版本,下图可见Torch所支持的CUDA最新版本位 11.7,如果我们安装CUDA 12.0 版本 执行 `torch.cuda.is_available()
` 时则为False。
PyTorch版本查询链接: Start Locally | PyTorch
2、下载安装
根据上图所示,选择合适的CUDA版本,安装CUDA C盘需要5 GB左右的空间,如果C盘空间少可使用分区助手来扩容,详细的安装过程不在概述,傻瓜试一键安装。
CUDA下载链接:CUDA Toolkit Archive | NVIDIA Developer
二、安装PyTorch
1、复制 `Run this Command` 这里的命令,在CMD中执行即可
标签:Python,Torch,避坑,PyTorch,CUDA,版本,安装 From: https://www.cnblogs.com/hlikex/p/16983707.html