测试 pycharm 是否成功安装配置 torch(pytorch)
·
1.新建IDE项目--->新建python文件,如图:

2.测试源码
import torch
print(torch.__version__)
print(torch.cuda.is_available())
x=torch.randn(1)
if torch.cuda.is_available():
device=torch.device("cuda")
y=torch.ones_like(x,device=device)
x=x.to(device)
z=x+y
print(z)
print(z.to("cpu",torch.double))
3. 显示成功

windows安装pytorch配置pycharm和Anaconda本人经验贴
https://blog.csdn.net/WJ_MeiMei/article/details/84141521
更多推荐

所有评论(0)