logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

Pytorch默认安装CPU版本(安装GPU版pytorch的坑)

今天在重装我的pytorch时,发现按部就班的操作后,检验torch.cuda.is_available()语句始终返回False。很奇怪,依照一般方法,我去检查了我的显卡驱动,重装了我的CUDA,检验了显卡驱动和cuda和pytorch版本号之间的联系,都没有发现问题。一筹莫展之际,我用conda list命令查看了我所安装的pytorch,发现居然是cpu版本,包括torchvision等等,

#pytorch
cuda中float2强制类型转换

记录一下,cuda初学者,遇到了将一个float类型的指针强制转换成float2类型指针的情况,不太明白是怎么进行转换的,遂做了一个小实验#include<cuda.h>#include<cuda_runtime.h>#include<iostream>using std::cout;using std::endl;int main(){float* arr1

#c++
到底了