pytorch学习记录
记录自己pytorch的学习过程
coding小白
这个作者很懒,什么都没留下…
展开
-
python将cifar10数据集转成图片格式的训练集和测试集
电脑环境:Python3.7需要注意路径,filename的那个地方~下面是训练集转化代码:import osfrom imageio import imsavedef unpickle(file): import pickle with open(file, 'rb') as fo: dict = pickle.load(fo, encoding='bytes') return dictfilename = './data/cifar-10-ba原创 2021-06-09 17:46:59 · 670 阅读 · 5 评论 -
服务器多卡训练pytorch程序
参考链接:https://muzhan.blog.csdn.net/article/details/109318226亲测有效~原创 2021-05-29 20:24:03 · 511 阅读 · 0 评论 -
服务器安装pytorch-gpu版本踩过的坑
1、安装显卡驱动那些, 很多其他的博文里面都有这一部分,我感觉大部分人遇到的情况应该是基本都安装好了,注意的是CUDA 和CUDNN的版本一定要对应,不然后面运行代码的时候,会出现各种各样的错误,比较常见的是那个 CUDNN ERROR~~2、安装anaconda 基本大家都是从这一步开始的,我推荐的方法是现在windows上下载好,再传到Linux服务器上,免得因为 网络的问题一直下载不了,anaconda的具体安装方法也有很多的博文来解释,基本就是一路回车就行了。3、换源原创 2021-05-19 16:23:41 · 618 阅读 · 0 评论