深度学习库,自己了解下
安装
Windows 安装
我试了几次 pip3 安装
我的版本
- Windows 7 64bit
- python36 / pip3
- torch-0.4.0
- 未安装CUDA
1 | pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp36-cp36m-win_amd64.whl |
装完以后 都不能 import torch
python 下导入1
2
3
4
5
6
7
8import torch
Traceback (most recent call last):
File "tensor01.py", line 5, in <module>
import torch
File "d:\app\python36\lib\site-packages\torch\__init__.py", line 78, in <modul
e>
from torch._C import *
ImportError: DLL load failed: 找不到指定的模块。
最后不得不放弃了,不知道是什么DLL找不到
Docker 安装
前面省略Docker 安装步聚
1 | docker pull pytorch/pytorch |
Linux/Mac 未曾试过
不知道能否成功