GPU版pytorch安装(win/linux)
参考:
Pytorch环境配置——cuda、、cudnn、torch、torchvision对应版本(最全)及安装方法-CSDN博客
Previous PyTorch Versions | PyTorch
法1:命令安装
如:
conda install pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-cuda=12.1 -c pytorch -c nvidia
法2:下载安装
https://download.pytorch.org/whl/torch_stable.html