我可以安装 pytorch cpu + 任何指定版本的 cudatoolkit 吗?
Can I install pytorch cpu + any specified version of cudatoolkit?
我的遥控器有 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。
我的问题是:以上两种安装方式本质上是一样的吗?如果没有,我如何安装 pytorch=1.10.0
和 cuda==11.0
?
我还想知道 cuda
兼容性如何工作? cudatoolkit==11.1
是否与使用 cudatoolkit==11.0
编译的程序兼容?
这完全取决于 pytorch 频道是否针对特定 cudatoolkit
版本构建了一个版本。我不知道具体的搜索方式,但可以在 pytorch 频道上 browse what builds are available。对于 linux-64 平台上的 PyTorch 1.10,似乎只有 CUDA 版本 10.2、11.1 和 11.3 可用。
如评论中所述,可以尝试使用
强制 PyTorch 的 CUDA 构建
conda create -n foo -c pytorch -c conda-forge cudatoolkit=11.0 'pytorch=*=*cuda*'
在这个组合中会失败。
至于兼容性,不,pytorch
包在 cudatoolkit
的次要版本中构建锁。例如,
我的遥控器有 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。
我的问题是:以上两种安装方式本质上是一样的吗?如果没有,我如何安装 pytorch=1.10.0
和 cuda==11.0
?
我还想知道 cuda
兼容性如何工作? cudatoolkit==11.1
是否与使用 cudatoolkit==11.0
编译的程序兼容?
这完全取决于 pytorch 频道是否针对特定 cudatoolkit
版本构建了一个版本。我不知道具体的搜索方式,但可以在 pytorch 频道上 browse what builds are available。对于 linux-64 平台上的 PyTorch 1.10,似乎只有 CUDA 版本 10.2、11.1 和 11.3 可用。
如评论中所述,可以尝试使用
强制 PyTorch 的 CUDA 构建conda create -n foo -c pytorch -c conda-forge cudatoolkit=11.0 'pytorch=*=*cuda*'
在这个组合中会失败。
至于兼容性,不,pytorch
包在 cudatoolkit
的次要版本中构建锁。例如,