本文主要是介绍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
这篇关于GPU版pytorch安装(win/linux)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!