1、检查在命令行模式下是否正确安装PyTorch
  • 打开 Anaconda prompt命令窗

  • 输入 命令 conda activate PyTorch,激活环境

  • 输入命令 python,进入python开发环境

  • 输入

    import torch
    torch.cuda.is_available()
    
  • 若正确安装PyTorch则会出现结果:True。这证明计算机上已经安装成功PyTorch,只是VScode没有正确导入而已。具体步骤请参考第二步。

  • 若没有出现此结果,证明Pytorch没有正确安装。

2、VScode中解决

这可能是由于安装的路径和VScode中访问路径不一致造成的。需要我们手动更改路径。

  • 首先保证VScode中安装了python扩展包。

  • 打开左上角File(文件)–>preferences(首选项)–>Settings(设置)

  • 在Extensions(扩展)中找到python

  • 点击“Edit in settings.json”

  • 打开后如下图,若不一样也没有关系。我们所要关注的就是图下红框所勾画出来的代码。若没有可以自行加上。其中将黄色字体更改为你的torch中python.exe的存在环境即可,最后记得保存设置文件、重新启动VScode。(有时候不会马上有反应,若还是报错,保存文件后可以重启电脑试试)

    image-20220509154637007

    若不知道自己的python路径,请继续往下看。

3、如何查看python路径。
  • 首先按照 第一步,成功在cmd中成功进入到python环境中。再输入一下代码。

    import sys
    print(sys.path)
    
  • 具体结果如下

    image-20220509154415427

  • 我们发现蓝色框中的部分就是大概我们要找的python的位置,所以自己去PyTorch文件夹中找到即可。

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐