我可以安装pytorch +任何指定版本的cudatoolkit吗？
我可以安装pytorch +任何指定版本的cudatoolkit吗？
提问于 2021-12-13 06:16:18
我的遥控器有cuda==11.0，我想在上面安装pytorch。
我使用命令conda install pytorch cudatoolkit=11.0 -c pytorch -c conda-forge，但在安装列表中：
cudatoolkit conda-forge/linux-64::cudatoolkit-11.0.3-h15472ef_8
pytorch pytorch/linux-64::pytorch-1.10.0-py3.8_cpu_0我发现pytorch是一个cpu 。
或者，我将11.0替换为11.1，安装列表看起来如下：
cudatoolkit conda-forge/linux-64::cudatoolkit-11.1.1-h6406543_8
pytorch pytorch/linux-64::pytorch-1.10.0-py3.8_cuda11.1_cudnn8.0.5_0其中pytorch是gpu one。
我的问题是：这两种安装本质上是相同的吗？如果没有，如何使用pytorch=1.10.0安装cuda==11.0？
我还想知道cuda兼容性是如何工作的？cudatoolkit==11.1与用cudatoolkit==11.0编译的程序兼容吗？
回答 1
Stack Overflow用户
回答已采纳
发布于 2021-12-13 16:06:19
这一切都取决于py手电筒通道是否针对特定的cudatoolkit版本构建了一个版本。我不知道具体的搜索方法，但是可以在浏览可用的构建的频道上搜索。对于PyTorch 1.10在linux-64平台上，似乎只有10.2、11.1和11.3版本可用。
正如注释中提到的，您可以尝试使用
conda create -n foo -c pytorch -c conda-forge cudatoolkit=11.0 'pytorch=*=*cuda*'在这个组合中会失败。
至于兼容性，不是，pytorch包在cudatoolkit的次要版本中构建了锁。例如,
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/70330604
复制相关文章
相似问题
腾讯云开发者
