使用pytorch进行人工智能实战过程常用语句 &一些问题

pytorch中文官方文档

pytorch0.2
pytorch0.3
pytorch0.4
pytorch1.0
pytorch1.4
pytorch1.7

pytorch英文文档

torch.rand系列

torch.rand()、torch.randn()、torch.randint()、torch.randperm()用法

nn.BatchNorm2d函数

NN.BATCHNORM2D()函数

torch.nn.dropout和torch.nn.dropout2d

torch.nn.dropout和torch.nn.dropout2d的区别
dropout随机使一个元素为0
dropout2d随机使一整个channel为0

nvidia-smi has failed because it couldn’t communicate with the NVIDIA driver.

显卡驱动报错：

cuda

查看cuda版本命令:nvcc -V

CUDA 及其驱动兼容情况

torch.cuda.is_available()一直返回False

DataFrame获取列名

1. df.columns
   返回的是array格式。
2. list(df)
   返回list格式。(最近获知，非常方便，原理待探究）
3. df.keys()

参考 【Python】DataFrame获取列名

保存模型(整个模型)

torch.save(网络实例化名, 'run.pt')
如果想要在每一轮训练后保存一次模型，就在训练的for循环里面的最后一行保存，但是每次保存的名字得是不一样的