深度学习
Vincent__Lai
这个作者很懒,什么都没留下…
展开
-
【Pytorch】快速输出网络模型信息方法
首先,安装torchsummarypip isntall torchsummary下面是demo代码,其中(3, 100, 100)分别为单张图片的通道数、高、宽from torchsummary import summaryimport torchvisionimport torchdef model_info(model): device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') back原创 2020-12-28 10:51:36 · 1753 阅读 · 1 评论 -
【Pytorch】 Attempting to deserialize object on a CUDA device but torch.cuda.is_available()
torch.loads原创 2020-08-18 10:50:21 · 2674 阅读 · 0 评论 -
【Pytorch】RuntimeError: one of the variables needed for gradient computation has been modified by
参考:RuntimeError: one of the variables needed for gradient computation has been modified by an inplace o报错RuntimeError: one of the variables needed for gradient computation has been modified by an in...原创 2020-04-06 15:35:36 · 742 阅读 · 0 评论 -
【深度学习】卷积层与池化层的参数计算
转载自:https://blog.csdn.net/qian99/article/details/79008053若为1个553的卷积核,则为1层:若为6个553的卷积核,则为6层:计算公式:池化层公式与卷积一样,只是不需要参数,例如MAX POOLING:...转载 2019-04-07 19:42:33 · 5798 阅读 · 0 评论 -
【CUDA】显卡驱动版本、CUDA、cudnn之间关系以及ubuntu18.04卸载安装多版本CUDA
一行命令卸载:https://blog.csdn.net/qq_33200967/article/details/80689543#CUDA_71ubuntu18.04安装CUDA9.0:https://www.jianshu.com/p/288c536a1092多版本切换:https://blog.csdn.net/qq_23996885/article/details/90205937...原创 2019-07-07 09:11:07 · 2280 阅读 · 1 评论 -
【CUDA】更改cuda版本后nvcc -V依然显示更改前的版本解决办法
问题描述电脑有两个版本的cuda,一个是cuda-8.0,一个是cuda-9.0,若原版本是cuda-8.0,nvcc -V后会显示:nvcc: NVIDIA ® Cuda compiler driverCopyright © 2005-2016 NVIDIA CorporationBuilt on Tue_Jan_10_13:22:03_CST_2017Cuda compilatio...原创 2019-10-01 16:22:08 · 24867 阅读 · 8 评论