安装的一些命令之类的,我就不重复了,只把最精华的部分说出来,尽量保持以下所说方法安装。
前提步骤:安装anaconda+环境变量配置。
找到自己电脑cuda版本,安装对应的cuda和cudnn(百度教程),安装的版本只能低,不许高。安装后,检查是否安装成功。
打开anconda,创建虚拟环境,命名随意(建议命名pytorch),python版本匹配你安装的cuda版本即可,创建好环境之后进行激活。
-
进入该环境下,一定要用conda命令更新numpy和pip,否则之后安装会报一系列莫名其妙的错误。先添加国内镜像,否则更新很慢,百度教程即可。现在在环境下conda list看一下包应该如下图所示:(我当时没有先装numpy。记住一定要安装numpy,否则就和我后面一样,安装pytorch报错,numpy)
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple numpy
最关键的部分到了,要下载对应pytorch的包了,我们这采用最简单的方法下载,其他的任何方法都不靠谱!!!直接确定好版本,上官网下载,速度不会很慢,但不会出错,记住,前面写了cpu的是只含有cpu,没写的就是gpu版本。
,然后自己pip install,这里我下载的版本的是(pytorch) F:\software>pip install torch-1.2.0-cp36-cp36m-win_amd64.whl
。结果发生报错,说要先装numpy,导致安装过程中系统自动安装numpy,但是此时是没有镜像的,速度很慢,导致爆红。如果已经安装了numpy,是不会报错的。
所以一定要安装numpy并且添加镜像下载。然后重新安装pytorch即可!
即成功!