使用pip install torch
,命令安装torch库,提示如下错误:
File "D:\soft\环境变量\TEMP\pip-install-xytf4pyb\torch\setup.py", line 51, in run
from tools.nnwrap import generate_wrappers as generate_nn_wrappers
ModuleNotFoundError: No module named 'tools.nnwrap'
最后在PyTorch官网利用其提供的安装命令,成功安装torch库。
根据自己的电脑配置选择安装命令
pip安装
pip install torch===1.5.1 torchvision===0.6.1 -f https://download.pytorch.org/whl/torch_stable.html
成功安装
Looking in links: https://download.pytorch.org/whl/torch_stable.html
Collecting torch===1.5.1
Downloading https://download.pytorch.org/whl/cu102/torch-1.5.1-cp36-cp36m-win_amd64.whl (913.8MB)
|████████████████████████████████| 913.8MB 8.7kB/s
WARNING: Retrying (Retry(total=4, connect=None, read=None, redirect=None, status=None)) after connection broken by 'ReadTimeoutError("HTTPSConnectionPool(host='pypi.org', port=443): Read timed out. (read timeout=15)",)': /simple/torchvision/
Collecting torchvision===0.6.1
Downloading https://download.pytorch.org/whl/cu102/torchvision-0.6.1-cp36-cp36m-win_amd64.whl (1.2MB)
|████████████████████████████████| 1.2MB 547kB/s
Requirement already satisfied: future in d:\soft\python3.6\lib\site-packages (from torch===1.5.1) (0.16.0)
Requirement already satisfied: numpy in d:\soft\python3.6\lib\site-packages (from torch===1.5.1) (1.16.6)
Requirement already satisfied: pillow>=4.1.1 in d:\soft\python3.6\lib\site-packages (from torchvision===0.6.1) (6.2.0)
Installing collected packages: torch, torchvision
Successfully installed torch-1.5.1 torchvision-0.6.1
参考:https://blog.csdn.net/cxyj666/article/details/97617358