Did you conda installed all the listed dependencies on the readme? It is very likely that the package is included in there as well.
conda installed numpy pyyaml mkl setuptools cmake cffi. it ran smoothly.
1 Like
I have a problem when using source installation，
after run ：
$ python setup.py install
These errors occurred：
src/common_coll.h:101:1: error: ‘struct KernelArgs’ has no member named ‘ring’
make[3]: *** [/home/zl/pytorch/torch/lib/build/nccl/obj/all_reduce.o] Error 1
make[2]: *** [lib/libnccl.so] Error 2
make[1]: *** [CMakeFiles/nccl.dir/all] Error 2
I also try use Commands via pip on the website ,It can be installed successfully，but can‘t use cudnn to speed up. my cudnn version is 6.0
and I print torch.version.cuda=8.0.61, torch.backends.cudnn.version()=7003.
I do not know where is wrong，what should I do？ thanks
What does it say when you try to use cudnn?