一、安装CUDA
CPU版本不需要安装CUDA。
CUDA官网地址:CUDA Toolkit 11.7 Update 1 Downloads | NVIDIA Developer
根据自己电脑的系统、版本等进行选择。我的是Windows 11 64位, 点击下载。
将下载好的exe文件双击,按步骤操作,进行安装
二、安装PyTorch
完成CUDA的安装再来下载PyTorch. PyTorch官方网站:PyTorch
CPU版本不需要安装CUDA。
CUDA官网地址:CUDA Toolkit 11.7 Update 1 Downloads | NVIDIA Developer
根据自己电脑的系统、版本等进行选择。我的是Windows 11 64位, 点击下载。
将下载好的exe文件双击,按步骤操作,进行安装
完成CUDA的安装再来下载PyTorch. PyTorch官方网站:PyTorch