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

 

 

 

 

Logo

一站式虚拟内容创作平台,激发创意,赋能创作,进入R空间,遇见同道,让优质作品闪耀发光。​

更多推荐