由于我们刚刚安装了miniconda3的Conda环境,里面安装了Python3.7和Pytorch
于是要进行测试一下,先试试用 PyCharm导入,以后再用 jupter
一、具体步骤
1、依次进入Preferences | Project: 项目名 | Python Interpreter
点击 show All
2、点击加号,点击添加本地解释器 Add Local Interpreter...
3、点击 Conda Environment
选择已有环境,并选择我们之前安装的 pytorch2.2.1虚拟环境
4、再次回到第1个页面,把这个勾上,让当前项目生效
5、最终结果如下
二、测试程序
新建一个py文件,运行之,能出结果就行
import torch
N, D = 3, 4
x = torch.tensor(torch.rand(N, D), requires_grad=True)
y = torch.tensor(torch.rand(N, D), requires_grad=True)
z = torch.tensor(torch.rand(N, D), requires_grad=True)
a = x * y
b = a + z
c = torch.sum(b)
c.backward()
print(x)
print(y)
print(z)
print(c)
您可以选择一种方式赞助本站
支付宝扫一扫赞助
微信钱包扫描赞助
赏