1.下载想要安装的cudnn下载链接：2.下载完成后解压所下载的cudnn压缩包bashtarxvfcudnn10.1osxx64v7.6.2.24.tgz3.复制文件bashsudocpcuda/include//usr/local/cuda/includesudocpcuda/lib//usr/local/cuda/lib64注：cp目录需要根据实际进行修改。

打开/创建文件bashvitest.py打开后进行修改注：下面的命令需要使用英文输入法。1、需要按行快速移动光标时，可以使用键盘上的编辑键Home，快速将光标移动至当前行的行首。除此之外，也可以在命令模式中使用快捷键"^"（即Shift6）或0（数字0）。2、如果要快速移动光标至当前行的行尾，可以使用编辑键End。也可以在命令模式中使用快

环境：python3、torch1.2.0(torch1.5.0也可以)、torchvision0.4.0代码：（注：DOWNLOADMNISTFalse第一次应该设为True，原因在代码注释里）pythonimporttorchfromtorch.autogradimportVariableimporttorch.utils.dataa

pythondefpartition(arr,low,high):ilow1pivotarrhighforjinrange(low,high):ifarrj<pivot:ii1arri,arrjarrj,arriarri1,arrhigharrhigh