AssertionError: Torch not compiled with CUDA enabled

[fengxinxin]

`

Name Version Build Channel

_anaconda_depends 2020.02 py37_0
alabaster 0.7.12 py37_0
anaconda custom py37_1
anaconda-client 1.7.2 py37_0
anaconda-project 0.8.4 py_0
applaunchservices 0.2.1 py_0
appnope 0.1.0 py37_0
appscript 1.1.0 py37h1de35cc_0
argh 0.26.2 py37_0
asn1crypto 1.3.0 py37_0
astroid 2.3.3 py37_0
astropy 4.0.1.post1 py37h1de35cc_0
atomicwrites 1.3.0 py37_1
attrs 19.3.0 py_0
autopep8 1.4.4 py_0
babel 2.8.0 py_0
backcall 0.1.0 py37_0
backports 1.0 py_2
backports.shutil_get_terminal_size 1.0.0 py37_2
beautifulsoup4 4.9.0 py37_0
bitarray 1.2.1 py37h1de35cc_0
bkcharts 0.2 py37_0
blas 1.0 mkl
bleach 3.1.4 py_0
blosc 1.16.3 hd9629dc_0
bokeh 2.0.1 py37_0
boto 2.49.0 py37_0
bottleneck 1.3.2 py37h776bbcc_0
bzip2 1.0.8 h1de35cc_0
ca-certificates 2020.4.5.1 hecc5488_0 conda-forge
certifi 2020.4.5.1 py37_0
cffi 1.14.0 py37hb5b8e2f_0
chardet 3.0.4 py37_1003
click 7.1.1 py_0
cloudpickle 1.4.0 py_0
clyent 1.2.2 py37_1
colorama 0.4.3 py_0
contextlib2 0.6.0.post1 py_0
cryptography 2.8 py37ha12b0ac_0
cudatoolkit 9.0 h41a26b3_0 anaconda
cudatoolkit-dev 10.1.243 h0b31af3_3 conda-forge
curl 7.69.1 ha441bb4_0
cycler 0.10.0 py37_0
cython 0.29.17 py37h0a44026_0
cytoolz 0.10.1 py37h1de35cc_0
dask 2.15.0 py_0
dask-core 2.15.0 py_0
dbus 1.13.12 h90a0687_0
decorator 4.4.2 py_0
defusedxml 0.6.0 py_0
diff-match-patch 20181111 py_0
distributed 2.15.0 py37_0
docutils 0.16 py37_0
entrypoints 0.3 py37_0
et_xmlfile 1.0.1 py37_0
expat 2.2.6 h0a44026_0
fastcache 1.1.0 py37h1de35cc_0
filelock 3.0.12 py_0
flake8 3.7.9 py37_0
flask 1.1.2 py_0
freetype 2.9.1 hb4e5f40_0
freetype-py 2.1.0.post1 pypi_0 pypi
fsspec 0.7.1 py_0
future 0.18.2 py37_0
get_terminal_size 1.0.0 h7520d66_0
gettext 0.19.8.1 h15daf44_3
gevent 1.4.0 py37h1de35cc_0
glib 2.63.1 hd977a24_0
glob2 0.7 py_0
gmp 6.1.2 hb37e062_1
gmpy2 2.0.8 py37h6ef4df4_2
greenlet 0.4.15 py37h1de35cc_0
h5py 2.10.0 py37h3134771_0
hdf5 1.10.4 hfa1e0ec_0
heapdict 1.0.1 py_0
html5lib 1.0.1 py37_0
hypothesis 5.8.3 py_0
icu 58.2 h0a44026_3
idna 2.9 py_1
imageio 2.8.0 py_0
imageio-ffmpeg 0.4.1 pypi_0 pypi
imagesize 1.2.0 py_0
importlib_metadata 1.5.0 py37_0
intel-openmp 2019.4 233
intervaltree 3.0.2 py_0
ipykernel 5.1.4 py37h39e3cac_0
ipython 7.13.0 py37h5ca1d4c_0
ipython_genutils 0.2.0 py37_0
ipywidgets 7.5.1 py_0
isort 4.3.21 py37_0
itsdangerous 1.1.0 py37_0
jbig 2.1 h4d881f8_0
jdcal 1.4.1 py_0
jedi 0.15.2 py37_0
jinja2 2.11.2 py_0
joblib 0.14.1 py_0
jpeg 9b he5867d9_2
json5 0.9.4 py_0
jsonschema 3.2.0 py37_0
jupyter 1.0.0 py37_7
jupyter_client 6.1.2 py_0
jupyter_console 6.1.0 py_0
jupyter_core 4.6.3 py37_0
jupyterlab 1.2.6 pyhf63ae98_0
jupyterlab_server 1.1.0 py_0
keyring 21.1.1 py37_2
kiwisolver 1.1.0 py37h0a44026_0
krb5 1.17.1 hddcf347_0
lazy-object-proxy 1.4.3 py37h1de35cc_0
libarchive 3.3.3 h786848e_5
libcurl 7.69.1 h051b688_0
libcxx 4.0.1 hcfea43d_1
libcxxabi 4.0.1 hcfea43d_1
libedit 3.1.20181209 hb402a30_0
libffi 3.2.1 h0a44026_6
libgfortran 3.0.1 h93005f0_2
libiconv 1.16 h1de35cc_0
liblief 0.9.0 h2a1bed3_2
libpng 1.6.37 ha441bb4_0
libsodium 1.0.16 h3efe00b_0
libspatialindex 1.9.3 h0a44026_0
libssh2 1.9.0 ha12b0ac_1
libtiff 4.1.0 hcb84e12_0
libxml2 2.9.9 hf6e021a_1
libxslt 1.1.33 h33a18ac_0
llvm-openmp 4.0.1 hcfea43d_1
llvmlite 0.32.0 py37h8c7ce04_0
locket 0.2.0 py37_1
lxml 4.5.0 py37hef8c89e_0
lz4-c 1.8.1.2 h1de35cc_0
lzo 2.10 h1de35cc_2
markupsafe 1.1.1 py37h1de35cc_0
matplotlib 3.1.3 py37_0
matplotlib-base 3.1.3 py37h9aa3819_0
mccabe 0.6.1 py37_1
mistune 0.8.4 py37h1de35cc_0
mkl 2019.4 233
mkl-service 2.3.0 py37hfbe908c_0
mkl_fft 1.0.15 py37h5e564d8_0
mkl_random 1.1.0 py37ha771720_0
mock 4.0.2 py_0
more-itertools 8.2.0 py_0
moviepy 1.0.2 pypi_0 pypi
mpc 1.1.0 h6ef4df4_1
mpfr 4.0.1 h3018a27_3
mpmath 1.1.0 py37_0
msgpack-python 1.0.0 py37h04f5b5a_1
multipledispatch 0.6.0 py37_0
nbconvert 5.6.1 py37_0
nbformat 5.0.4 py_0
ncurses 6.2 h0a44026_1
networkx 2.3 pypi_0 pypi
ninja 1.9.0 py37h04f5b5a_0
nltk 3.4.5 py37_0
nose 1.3.7 py37_2
notebook 6.0.3 py37_0
numba 0.49.0 py37h6c726b0_0
numexpr 2.7.1 py37hce01a72_0
numpy 1.18.1 py37h7241aed_0
numpy-base 1.18.1 py37h6575580_1
numpydoc 0.9.2 py_0
olefile 0.46 py37_0
opencv-python 4.2.0.32 pypi_0 pypi
openpyxl 3.0.3 py_0
openssl 1.1.1g h0b31af3_0 conda-forge
packaging 20.3 py_0
pandas 1.0.3 py37h6c726b0_0
pandoc 2.2.3.2 0
pandocfilters 1.4.2 py37_1
parso 0.5.2 py_0
partd 1.1.0 py_0
path 13.1.0 py37_0
path.py 12.4.0 0
pathlib2 2.3.5 py37_0
pathtools 0.1.2 py_1
patsy 0.5.1 py37_0
pcre 8.43 h0a44026_0
pep8 1.7.1 py37_0
pexpect 4.8.0 py37_0
pickleshare 0.7.5 py37_0
pillow 7.0.0 py37h4655f20_0
pip 20.0.2 py37_1
pkginfo 1.5.0.1 py37_0
pluggy 0.13.1 py37_0
ply 3.11 py37_0
proglog 0.1.9 pypi_0 pypi
prometheus_client 0.7.1 py_0
prompt-toolkit 3.0.4 py_0
prompt_toolkit 3.0.4 0
psutil 5.7.0 py37h1de35cc_0
ptyprocess 0.6.0 py37_0
py 1.8.1 py_0
py-lief 0.9.0 py37h1413db1_2
pycodestyle 2.5.0 py37_0
pycosat 0.6.3 py37h1de35cc_0
pycparser 2.20 py_0
pycrypto 2.6.1 py37h1de35cc_9
pycurl 7.43.0.5 py37ha12b0ac_0
pydocstyle 4.0.1 py_0
pyflakes 2.1.1 py37_0
pygments 2.6.1 py_0
pylint 2.4.4 py37_0
pyodbc 4.0.30 py37h0a44026_0
pyopenssl 19.1.0 py37_0
pyparsing 2.4.6 py_0
pyqt 5.9.2 py37h655552a_2
pyrsistent 0.16.0 py37h1de35cc_0
pysocks 1.7.1 py37_0
pytables 3.6.1 py37h5bccee9_0
pytest 5.4.1 py37_0
pytest-arraydiff 0.3 py37h39e3cac_0
pytest-astropy 0.8.0 py_0
pytest-astropy-header 0.1.2 py_0
pytest-doctestplus 0.5.0 py_0
pytest-openfiles 0.4.0 py_0
pytest-remotedata 0.3.2 py37_0
python 3.7.7 hc70fcce_0_cpython
python-dateutil 2.8.1 py_0
python-jsonrpc-server 0.3.4 py_0
python-language-server 0.31.10 py37_0
python-libarchive-c 2.9 py_0
python.app 2 py37_10
python_abi 3.7 1_cp37m conda-forge
pytorch 1.4.0 py3.7_0 pytorch
pytz 2019.3 py_0
pywavelets 1.1.1 py37h1de35cc_0
pyyaml 5.3.1 py37h1de35cc_0
pyzmq 18.1.1 py37h0a44026_0
qdarkstyle 2.8.1 py_0
qt 5.9.7 h468cd18_1
qtawesome 0.7.0 py_0
qtconsole 4.7.3 py_0
qtpy 1.9.0 py_0
readline 8.0 h1de35cc_0
requests 2.23.0 py37_0
ripgrep 11.0.2 he32d670_0
rope 0.16.0 py_0
rtree 0.9.3 py37_0
ruamel_yaml 0.15.87 py37h1de35cc_0
scikit-image 0.16.2 py37h6c726b0_0
scikit-learn 0.22.1 py37h27c97d8_0
scipy 1.4.1 py37h9fa6033_0
seaborn 0.10.0 py_0
send2trash 1.5.0 py37_0
setuptools 46.1.3 py37_0
simplegeneric 0.8.1 py37_2
singledispatch 3.4.0.3 py37_0
sip 4.19.8 py37h0a44026_0
six 1.14.0 py37_0
snappy 1.1.7 he62c110_3
snowballstemmer 2.0.0 py_0
sortedcollections 1.1.2 py37_0
sortedcontainers 2.1.0 py37_0
soupsieve 2.0 py_0
sphinx 3.0.3 py_0
sphinxcontrib 1.0 py37_1
sphinxcontrib-applehelp 1.0.2 py_0
sphinxcontrib-devhelp 1.0.2 py_0
sphinxcontrib-htmlhelp 1.0.3 py_0
sphinxcontrib-jsmath 1.0.1 py_0
sphinxcontrib-qthelp 1.0.3 py_0
sphinxcontrib-serializinghtml 1.1.4 py_0
sphinxcontrib-websupport 1.2.1 py_0
spyder 4.1.2 py37_0
spyder-kernels 1.9.0 py37_0
sqlalchemy 1.3.16 py37h1de35cc_0
sqlite 3.31.1 h5c1f38d_1
statsmodels 0.11.0 py37h1de35cc_0
sympy 1.5.1 py37_0
tbb 2020.0 h04f5b5a_0
tblib 1.6.0 py_0
terminado 0.8.3 py37_0
testpath 0.4.4 py_0
tk 8.6.8 ha441bb4_0
toolz 0.10.0 py_0
torchvision 0.5.0 py37_cpu pytorch
tornado 6.0.4 py37h1de35cc_1
tqdm 4.45.0 py_0
traitlets 4.3.3 py37_0
transforms3d 0.3.1 pypi_0 pypi
typing_extensions 3.7.4.1 py37_0
ujson 1.35 py37h1de35cc_0
unicodecsv 0.14.1 py37_0
unixodbc 2.3.7 h1de35cc_0
urllib3 1.25.8 py37_0
vispy 0.6.4 pypi_0 pypi
watchdog 0.10.2 py37h1de35cc_0
wcwidth 0.1.9 py_0
webencodings 0.5.1 py37_1
werkzeug 1.0.1 py_0
wheel 0.34.2 py37_0
widgetsnbextension 3.5.1 py37_0
wrapt 1.12.1 py37h1de35cc_1
wurlitzer 2.0.0 py37_0
xlrd 1.2.0 py37_0
xlsxwriter 1.2.8 py_0
xlwings 0.18.0 py37_0
xlwt 1.3.0 py37_0
xz 5.2.5 h1de35cc_0
yaml 0.1.7 hc338f04_2
yapf 0.28.0 py_0
zeromq 4.3.1 h0a44026_3
zict 2.0.0 py_0
zipp 2.2.0 py_0
zlib 1.2.11 h1de35cc_3
zstd 1.3.7 h5bba6e5_0
`

[fengxinxin]

done

[unknownArtist]

Any update regarding this issue..
I am on mac.. and facing same issue when running command python main.py --config argument.yml

[unknownArtist]

@fengxinxin i overcome this issue after setting value gpu_ids: -1 in argument.yml