使用conda安装支持GPU的旧版本pytorch

ncecgwcz 于 2023-01-09 发布在其他

关注(0)|答案(1)|浏览(218)

我想使用conda和gpu support为特定的cuda版本安装pytorch1.11，例如11.6。
可以执行以下操作来安装最新版本：

conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia

HERE说明了如何在没有gpu支持的旧版本pytorch中执行此操作：

conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 -c pytorch

那么，下面的代码是否使用conda安装了一个带有gpu support的pytorch 1.11？

conda install pytorch==1.11.0 torchvision==0.12.0 pytorch-cuda=11.6 -c pytorch -c nvidia

pytorch

来源：https://stackoverflow.com/questions/75017489/install-older-version-of-pytorch-with-gpu-support-using-conda

1条答案

按热度按时间

wfsdck301#

pytorch-cuda包是三周前才引入的一个新的元包--除了最新的PyTorch版本之外，它与任何东西都无关。相反，对于旧的包，您可以直接指定环境的cudatoolkit版本。
请注意，pytorch=1.11仅支持CUDA版本10.2和11.1-11.5。没有可用于11.6的构建版本。请参见Anaconda Cloud。

示例：CUDA 11.5

conda install pytorch=1.11.0 torchvision=0.12 torchaudio=0.11 cudatoolkit=11.5 -c pytorch -c conda-forge

赞(0）回复(0）举报 2023-01-09

我来回答

使用conda安装支持GPU的旧版本pytorch

1条答案

相关问题

热门标签

最新问答