frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Use conda torchvision version (#20865)

Browse Source 
Summary:
This tries to fix the following error on current master:
```
May 23 16:18:47 Traceback (most recent call last):
May 23 16:18:47   File "main.py", line 7, in <module>
May 23 16:18:47     from torchvision import datasets, transforms
May 23 16:18:47   File "/opt/conda/lib/python3.6/site-packages/torchvision/__init__.py", line 1, in <module>
May 23 16:18:47     from torchvision import models
May 23 16:18:47   File "/opt/conda/lib/python3.6/site-packages/torchvision/models/__init__.py", line 11, in <module>
May 23 16:18:47     from . import detection
May 23 16:18:47   File "/opt/conda/lib/python3.6/site-packages/torchvision/models/detection/__init__.py", line 1, in <module>
May 23 16:18:47     from .faster_rcnn import *
May 23 16:18:47   File "/opt/conda/lib/python3.6/site-packages/torchvision/models/detection/faster_rcnn.py", line 7, in <module>
May 23 16:18:47     from torchvision.ops import misc as misc_nn_ops
May 23 16:18:47   File "/opt/conda/lib/python3.6/site-packages/torchvision/ops/__init__.py", line 1, in <module>
May 23 16:18:47     from .boxes import nms, box_iou
May 23 16:18:47   File "/opt/conda/lib/python3.6/site-packages/torchvision/ops/boxes.py", line 2, in <module>
May 23 16:18:47     from torchvision import _C
May 23 16:18:47 ImportError: /opt/conda/lib/python3.6/site-packages/torchvision/_C.cpython-36m-x86_64-linux-gnu.so: undefined symbol: _ZN2at19NonVariableTypeMode10is_enabledEv
```
Pull Request resolved: https://github.com/pytorch/pytorch/pull/20865

Differential Revision: D15481736

Pulled By: yf225

fbshipit-source-id: 67d4fd70652ccc709b44cb15392d6e44a8fe9235
This commit is contained in:
Will Feng
2019-05-23 13:39:37 -07:00
committed by Facebook Github Bot
parent b6d0f6c85a
commit 2d96876d88
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.jenkins/pytorch/perf_test/test_cpu_speed_mnist.sh
View File

@ -12,7 +12,7 @@ test_cpu_speed_mnist () {
cd examples/mnist
pip install -q -r requirements.txt
conda install -c pytorch torchvision-cpu
# Download data
python main.py --epochs 0

2
.jenkins/pytorch/perf_test/test_gpu_speed_mnist.sh
View File

@ -12,7 +12,7 @@ test_gpu_speed_mnist () {
cd examples/mnist
pip install -q -r requirements.txt
conda install -c pytorch torchvision
# Download data
python main.py --epochs 0