代码之家  ›  专栏  ›  技术社区  ›  questionto42

如果我更新了anaconda cudatoolkit,pytorch(安装命令的cudatoolkit-max版本10.2)会受到伤害吗?

  •  0
  • questionto42  · 技术社区  · 6 年前

    一些问题来自 https://superuser.com/questions/1572640/do-i-need-to-install-cuda-separately-after-installing-the-nvidia-display-driver .其中一个问题:

    conda-pytorch需要与官方的非conda/非pip cuda工具包不同的版本吗 https://developer.nvidia.com/cuda-toolkit ?

    换句话说:我可以使用NVIDIA的“cuda工具包”来安装pytorch吗?

    背景:

    如果你通过“命令助手” https://pytorch.org/get-started/locally/ ,您可以在cuda版本9.2、10.1、10.2和None之间进行选择。

    取10.2可得到:

    conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
    

    使用“None”可以构建以下命令,但您也不能在pytorch中使用cuda:

    conda install pytorch torchvision cpuonly -c pytorch
    

    然后,我可以使用NVIDIA“cuda toolkit”10.2版作为conda cudatoolkit,使此命令与使用以下命令执行时相同吗 cudatoolkit=10.2 参数?

    由于pytorch安装了不同的版本(10.2而不是最新的NVIDIA 11.0),并且conda安装需要额外的325 MB,因此出现了这个问题。如果两个版本都是11.0,安装尺寸较小,您甚至可能没有注意到可能的差异。但现在很明显,conda拥有自己的独立于NVIDIA的cuda版本。

    0 回复  |  直到 5 年前
        1
  •  2
  •   Robert Crovella    6 年前

    我想有可能让安装了conda的pytorch使用未安装conda的CUDA工具包。我不知道该怎么做,根据我的经验,当使用依赖于CUDA的conda包时,提供一个安装了conda的CUDA工具包并让它使用它要容易得多,而不是其他任何东西。这通常意味着我在conda中安装了一个CUDA工具包,另一个安装在通常的位置。

    然而,无论你如何安装pytorch,如果你安装了一个二进制包(例如通过conda),那么pytorch的版本将取决于CUDA的特定版本(例如10.2),以及 无论CUDA的安装方式或位置如何,都不能使用任何其他版本的CUDA来满足该依赖关系 .

        2
  •  1
  •   Mikhail Lobanov    5 年前

    您可以尝试通过Pip安装PyTorch:

    pip install torch torchvision
    

    这也是官方的安装方式,可以在“命令助手”中找到 https://pytorch.org/get-started/locally/ .

    它使用预装的CUDA,不下载自己的CUDA工具包。 您还可以选择CUDA的版本来安装PyTorch:

    pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html