前提条件:Anaconda环境、Pytorch已正确安装。没完成这一步或想验证是否安装成功的朋友可以按照“Anaconda安装”“Pytorch安装”等关键词搜索,b站up主我是土堆的Pytorch入门教程中P1视频也是蛮详细的。

问题起因:在b站看up主我是土堆的Pytorch入门教程中Python编辑器的配置环节,在控制台输入import torch时出现如下错误:

后来发现是新建项目时,由于和土堆老师pycharm版本不同于是一些选项出现了错误。话不多说直接展示正确选项:

比较关键的就是,Type选择Conda,Path to conda即conda路径根据你的情况选anaconda目录下的conda.bat,Environment即环境选择pytorch。如果conda路径你选对了那么环境的下拉框会有选项的,不会是不可选状态。

然后按照土堆老师视频里验证方法,控制台输入import torch没有报错,再输入torch.cpu.is_available()后返回了True,这就表示成功咯。

注:土堆老师教程是按照有英伟达显卡的情况,安装CUDA下Pytorch,他输入的是torch.coda.is_available()。

Logo

分享最新的 NVIDIA AI Software 资源以及活动/会议信息,精选收录AI相关技术内容,欢迎大家加入社区并参与讨论。

更多推荐