【Pytorch】使用星核pytorch环境

1、制作虚拟环境

星核的python环境在C:\envmain下,开发的时候不要直接在该环境下开发!!,以免污染主环境,而是应该使用该环境的python在你需要的文件夹目录中,制作一个.venv虚拟环境。具体方式如下:

打开cmd进入你想要制作虚拟环境的目录,例如c:\users\xh\wk

输入如下命令,制作你的虚拟环境:

c:\envmain\python -m venv --system-site-packages <your_dir>

2、使用虚拟环境

然后进入<your_dir>该目录(上图为torchtest2)

运行Scripts\activate以激活python环境,激活之后命令行前面会显示括号(虚拟环境的名字),如图

image

在该环境激活时,可以直接使用python命令

import torch之后就可以调用pytorch啦,示例如下:

如果出现“找不到指定模块”可以无视,这个是旧版rocm的信息,不影响使用

会了吗?哈哈