【问题标题】:cuda is not available on my pytorch, but I can't find anything wrong with the versioncuda 在我的 pytorch 上不可用,但我找不到版本有什么问题
【发布时间】:2021-03-03 03:21:04
【问题描述】:

出于某种原因,我必须使用 cuda version10.0 而不是升级它 DriverAPI 的版本高于 RunTimeAPI 但有人告诉我没关系 最后问同样问题的其他人发现他们的版本不匹配。 emmmm不喜欢我

这里有详细信息

操作系统:Windows10 Python 3.7&3.8 都试过了

“nvcc -V”的结果: nvcc:发布 10.0、V10.0.130

nvidia-smi: NVIDIA-SMI 451.67 驱动程序版本:451.67 CUDA 版本:11.0

conda 列表:cudatoolkit 10.0.130 0

进口火炬 print(torch.version.cuda)-----无 torch.cuda.is_available()-----False

【问题讨论】:

    标签: python pytorch


    【解决方案1】:

    你是怎么安装的?我假设用点子。对于 pytorch,我建议从 https://download.pytorch.org/whl/torch_stable.html 手动下载轮子并安装:

        pip install torch-1.4.0+cu100-cp38-cp38-linux_x86_64.whl
    

    假设 python 3.8 和 linux。如果您使用不同的东西,请确保为您的操作系统、Cuda 版本和 python 解释器选择合适的版本。

    【讨论】:

    • 我在 pytorch.org 上运行命令更新到 v11.0,现在一切正常。我以相同的方式安装 v10.0,只是从 cudatoolkit=10.0 到 cudatoolkit=11.0
    【解决方案2】:

    确保您的 Torch、cuda 和 os 版本与这个官方 Pytorch 工具兼容 - https://pytorch.org/get-started/locally/

    该工具将准确告诉您要安装哪个 pytorch 以及要运行的命令

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 2014-01-05
      • 1970-01-01
      • 1970-01-01
      • 2019-06-13
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2020-03-05
      相关资源
      最近更新 更多