全部
在conda虚拟环境上使用CUDA 11.2设置PyTorch 1.7.1之后,我运行python setup.py install
它总是向我返回以下错误消息。
Traceback (most recent call last):
File "setup.py", line 2, in <module>
from torch.utils.cpp_extension import BuildExtension, CUDAExtension
File "anaconda3/envs/pytorch/lib/python3.7/site-packages/torch/__init__.py", line 189, in <module>
_load_global_deps()
File "anaconda3/envs/pytorch/lib/python3.7/site-packages/torch/__init__.py", line 142, in _load_global_deps
ctypes.CDLL(lib_path, mode=ctypes.RTLD_GLOBAL)
File "anaconda3/envs/pytorch/lib/python3.7/ctypes/__init__.py", line 364, in __init__
self._handle = _dlopen(self._name, mode)
OSError: anaconda3/envs/pytorch/lib/python3.7/site-packages/torch/lib/../../../../libcublas.so.11: symbol free_gemm_select version libcublasLt.so.11 not defined in file libcublasLt.so.11 with link time reference
有人可以帮忙吗?谢谢!
最后,我仅使用pip
Pytorch官方网站上的查找解决方案。
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句