深度学习
VioletEvergarden丶
校招冲冲冲
展开
-
2022完整总结WIN11 Pytorch与GPU环境配置
2022一文搞定深度学习环境配置安装Python安装CUDA安装Anaconda安装Pytorch安装PythonPython官网选择版本进行下载,推荐3.9,最新的版本可能存在适配兼容性的问题安装CUDA仅限于NVIDIA显卡,桌面右击进入NVIDIA控制面板,左下角点击系统信息,查看显卡驱动版本根据显卡驱动版本选择CUDA版本显卡驱动版本支持文档注意安装的时候需要注意不能无脑安装最新版本,需要先看Pytorch支持的最新版本超过最新支持版本可能出现兼容性问题安装CUDA过程原创 2022-01-19 12:19:40 · 3837 阅读 · 6 评论 -
CNN中1*1卷积核与Pytorch实现
CNN中1*1卷积核的意义信道压缩信道压缩一个32* 6* 6的图片,32通道,6维矩阵图片当用32* 1* 1的的卷积核提取特征时,得到的是一个6*6的图片卷积运算中输出的通道数与卷积核的数量有关,但是计算过程中,图片的通道数要和卷积核的通道数相同此时,通过1维卷积核将原来图片在通道维度上进行压缩,即将通道信息进行融合1* 1也可以理解为一个全连接网络,分别作用所有通道上面总结:总结一下,1x1的卷积核可以进行降维或者升维,也就是通过控制卷积核(通道数)实现,这个可以帮助减少模型参原创 2021-05-14 16:40:04 · 2249 阅读 · 0 评论 -
PyTorch DNN实现MNIST手写数字识别
import torchimport torch.nn as nnimport torch.optim as optimfrom torchvision import datasets, transformsbatch_size = 200learning_rate = 0.01epochs = 10train_loader = torch.utils.data.DataLoader( datasets.MNIST('../data', train=True, download=T原创 2021-04-23 09:28:44 · 533 阅读 · 0 评论 -
如何用pycharm来查看PyTorch,CUDA版本
如何用python来查看pycharm,CUDA版本import torchprint(torch.__version__)print(torch.version.cuda)输出结果原创 2021-04-20 20:48:03 · 8950 阅读 · 0 评论