mathematican的博客

pytorch加载和保存模型

gcc, cudatoolkit匹配问题

conda 创建、删除envconda 创建环境命令conda create -n xxx python=3.7conda 删除环境命令conda remove -n xxx --all

配置conda环境

安装完深度学习环境后需要检查GPU是否可用，方法如下TensorFlowimport tensorflow as tfprint(tf.test.is_gpu_available())PyTorchimport torchprint(torch.cuda.is_available())

层归一化的调用命令：torch.nn.LayerNorm(normalized_shape, eps=1e-05, elementwise_affine=True, device=None, dtype=None)在CV中的应用案例Image Example N, C, H, W = 20, 5, 10, 10 input = torch.randn(N, C, H, W) # input data # Normalize over the last three dimensions (i.

使用最新的卡RTX3080，需要安装11.3 cudatoolkit

pytorch,tqdm,matplotlib,h5py,scikit-image

代码如下torch.FloatTensor((2,3,4)).normal_(mean=0, std=1)解释：初始化一个shape为(2,3,4)的网络参数，初始化值取自标准正态分布。