测试 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

 

 

 

 

  • 15
    点赞
  • 72
    收藏
    觉得还不错? 一键收藏
  • 8
    评论
评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值