windows10 安装 pytorch

转自:https://blog.csdn.net/a19990412/article/details/82752224


环境

windows 10
python 64位(只有3.5 3.6 3.7三个版本会支持

步骤
点击下面的链接,进行下载 pytorch的版本
https://www.lfd.uci.edu/~gohlke/pythonlibs/#pytorch

前面的两种上是需要安装cuda的。对于我们这种新手来说暂时还用不到这个版本。
我选的是第四个,因为我的Python版本是3.6的。
下载下来之后,通过命令行进入到下载的文件夹下

第一步,通过dir *.whl来得到具体的名字,之后方便直接复制
第二步,通过pip install + 上面第一步的名字。就可以啦~ 具体的操作如下。

PS D:\Chrome(DownLoad)> dir *.whl


    目录: D:\Chrome(DownLoad)


Mode                LastWriteTime         Length Name
----                -------------         ------ ----
-a----        2018/9/18      9:09       47619741 torch-0.4.1-cp36-cp36m-win_amd64.whl


PS D:\Chrome(DownLoad)> pip install torch-0.4.1-cp36-cp36m-win_amd64.whl
Processing d:\chrome(download)\torch-0.4.1-cp36-cp36m-win_amd64.whl
Installing collected packages: torch
Successfully installed torch-0.4.1
123456789101112131415
目前的话,就算是已经安装好了pytorch,但是如果在python中import这个包的话,就会有下面的这个错误。
>>> import torch
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "D:\SoftWare\Python\lib\site-packages\torch\__init__.py", line 80, in <module>
    from torch._C import *
ImportError: DLL load failed: 找不到指定的模块。
>>>
1234567
这时候 访问下面这个链接。
https://anaconda.org/anaconda/intel-openmp/files

在这里插入图片描述

比如说,我就是下载的第一行的那个win-64,这是取决于windows的版本(一般都选64),主要是之前我下载的pytorch也是64位的。
对于下载下来的这个文件,需要解压两次才行!!!。
之后进入到 intel-openmp-2019.0-117\Library\bin,这个目录下,把两个dll文件,放在另外一个已经被放在环境变量的文件夹下(对于新手来说,这里推荐放在,Python.exe所在的目录下~)

在这里插入图片描述

到这里,Pytorch已经安装完成了。重启一下Python就好了~
D:\>Python
Python 3.6.6 (v3.6.6:4cf1f54eb7, Jun 27 2018, 03:37:03) [MSC v.1900 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>>
12345
嘿嘿~
不过,网上推荐还要安装pytorchvision这个库~
这是pypi上这个库的发行website.
https://pypi.org/project/torchvision/0.1.8/
其实直接在命令行下输入下面的命令就好~
pip install torchvision
1
到这里,一般来说,已经安装完成了,如果你想要安装更高级的版本(比如说GPU版的)
你可能还需要安装cuda,这个可以搜索一下,(知乎上有蛮多解释的。)我这就不放了~
---------------------
作者:肥宅_Sean
来源:CSDN
原文:https://blog.csdn.net/a19990412/article/details/82752224
版权声明:本文为博主原创文章,转载请附上博文链接!

猜你喜欢

转载自blog.csdn.net/u014513863/article/details/85274762