Windows 10 + Anaconda下安装 Pytorch
最近想把机器学习又开始拾起来,再次比较了下 Tensor Flow 和 Pytorch,准备选一个用,看了下各种评论,最后选了 Pytorch,原因嘛,反正就是自己玩,比较简单,也不做工业部署,分布式这些,PyTorch看起来没那么复杂配置过程。其实以前用过 Keras+Tensor Flow的组合,问了下业内人士,Keras也有在用的,不过今天就试试 Pytorch吧。
去官网看看怎么安装吧:
本机有一块1660 Super的显卡,可以装 CUDA,去 Nvidia的网站下载:
https://developer.nvidia.com/zh-cn/cuda-downloads
这里我选择了11.1 的版本,这里有两个下载,一个是小一点的网络安装 exe文件(50M+),另外一个是完整的本地安装包(3G左右)。应该是cdn的原因吧,nv的这个下载网站巨慢,下载的网络安装文件运行后就基本没反应,最后是报错了,所以这里推荐下载本地安装包,我用迅雷的“玩客云”下载速度也还挺快的。
安装这里没什么可说的。
安装 Pytorch 这里的时候也遇到同样的麻烦,用 Anaconda 的包安装命令,看到下载界面后,就基本上没反应了。
去这里