前提

已经通过Anaconda Promp装好pytorch,并且测试成功,同时vscode也已经安装好。
在这里插入图片描述

配置环境变量

在这里插入图片描述

在这里插入图片描述
在两处path处都配置环境变量,配置方式如下在这里插入图片描述
然后点击确定。

vscode中配置json

打开Vscode,左上角点击文件>>首选项>>设置>>扩展,在扩展中找到python,点击右侧在这里插入图片描述
修改下图红色框中的路径,修改为安装在Anaconda中的pytorch中python.exe的位置,打开和我的不一样很正常,缺少的就将它补上。在这里插入图片描述

测试

记得保存文件,然后重启vscode。创建py文件运行如下代码:

import torch
print(torch.cuda.is_available())

运行成功

Logo

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

更多推荐