vs
still_learning
这个作者很懒,什么都没留下…
展开
-
pytorch 把tensor转换成int c++/libtorch
直接在tensor变量的后面加.item(),就能把tensor类型转换成at::scales类型之后调用toInt()即可。原创 2021-01-28 15:22:37 · 2548 阅读 · 0 评论 -
CMake常见配置
设置64位 vs2017参数cmake -G "Visual Studio 14 Win64"原创 2021-01-25 13:01:55 · 427 阅读 · 0 评论 -
libtorch,pytorch的c++版本如何查看是否可以使用cuda
#include <torch/torch.h>#include <iostream>int main() { torch::Tensor tensor = torch::rand({ 2, 3 }); std::cout << tensor << std::endl; std::cout << torch::cuda::is_available() << std::endl; std::cout << tor原创 2021-01-24 19:41:45 · 2510 阅读 · 1 评论 -
c++调用Pytorch,torchlib环境配置
windows系统+vs2017+cmake之前已经尝试过在QT上配置环境。但就算编译通过以后,仍然会出现闪退,错误返回值的情况,所以笔者尝试使用官方教程的方法,一步步构建VS2017的环境,期间也踩了一些坑,在这里总结一下和大家分享。参考教程:https://pytorch.org/cppdocs/installing.htmlCMake is an open-source, cross-platform family of tools designed to build, test and p原创 2021-01-24 15:40:27 · 1420 阅读 · 0 评论