admin管理员组

文章数量:1530046

 

现在的AMD显卡已经可以使用GPU,但我下载时选择的是CUDA10.2的版本。导入到pycharm的环境中后,运行torch.cuda.is_available返回Ture。但是运行terson.to(desive = "cuda")报错

表示计算机的算力是86,而10.2的版本最高只能支持到算力75, 所以需要在pytorch官网中下载最新版本11.1.。

但要注意不能直接复制

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge

下载。因为anaconda自带的conda不能下载安装。会在等待下载半天最后报错。我们需要下载anaconda的一个包叫conda-forge.并将此设置为最高通道。

因为我的pytorch在增加的虚拟环境中,所以我有base和pytorch两个环境,考虑我是在pytorch中安装cuda,所以先激活环境

本文标签: 报错torchCUDA