1.安装cuda
推荐参看这一篇,GPU版本pyTorch安装教程(实测,特别详细)
最关键的是作者最后的
重点了,下载GPU版本的torch和torvision,两都要下
这一部分,之前搜了不少文章,一直不成功,直到看到这里,才发现这部分是其他文章里没有的,做了之后才终于搞定。
话又说回来,用notebook之前我在独立python环境下安装pytorch+cuda都已经成功,不过这是好几个月之前的事情了,所以当时可能就已经按照另外的文档做过这一步,但是在anaconda下安装pytorch和cuda支持的时候根本就没注意到,毕竟在当时是全部使用pip安装的。
2.jupyter notebook 内核死掉的问题
这个问题是在用matplotlib画图的时候出现的,一调用ishow就会崩溃,经过各种排查,只导入matplotlib.pyplot
而不导入torch
的话,就一切正常,但不导入torch
是不可能的,经过各种搜索,发现notebook出现这个问题并不只是使用matplolib才会出现,解决的方法之一是pip install numpy --upgrade
,这就很神奇了,但是对我来说是有效的
3.关于mnist.pkl.gz"
github不太好上,所以从https://github.com/pytorch/tutorials/raw/master/_static/
获取这个文件有时候比较困难,不过可以从https://resources.oreilly.com/live-training/inside-unsupervised-learning/-/raw/9f262477e62c3f5a0aa7eb788e557fc7ad1310de/data/mnist_data/
这里下,速度还不错。
4.pycocotools 下载及安装
可以在这里下一个tar.gz的包。至于安装会有一些问题,值得研究一下。