python编程时的小细节

1.代码中出现中文，在该页代码最前面加上

#coding:utf-8

2. pytorch编程出现问题：

RuntimeError: cuDNN version mismatch: PyTorch was compiled against 7005 but linked against 7103

解决方法：通过搜索找到cudnn-7.1.3-cuda8.0_0的安装路径，在该路径下安装cudnn

conda install cudnn=7.0.5

3.采用命令行查看CUDNN版本：

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

4．多ＧＰＵ情况下，选择指定的ＧＰＵ进行训练：

CUDA_VISIBLE_DEVICES=0
torch.cuda.set_Device(0)

或

os.system('nvidia-smi -q -d Memory |grep -A4 GPU|grep Free >tmp')
memory_gpu = [int(x.split()[2]) for x in open('tmp','r').readlines()]
os.system('rm tmp')
os.environ["CUDA_VISIBLE_DEVICES"] = str(np.argmax(memory_gpu))