AlexCDD的博客

原创 pytorch 安装

3、完整的pytorch安装要包含torch，torchvision，torchaudio三件套，要注意的三件套的版本要对应匹配。如果不清楚版本对应关系，可以先下载torch的whl包，安装成功后。如果是在国内，下载可能会很慢。因此，直接下载对应版本的whl包，再离线安装。2、下载whl包时，要注意本地机器是否有nvidia显卡，如果没有只需下载cpu版本的包，如果有可以下载cuda版本的包。会默认下载与已安装torch的对应的torchvision和torchaudio版本。