本博客主要解决在torch使用中遇到的问题与解决~
1.安装相关问题
1.1.conda虚拟环境内无法安装torch(pip install torch ×)
解决方案:
如果是GPU版本,先查看cuda版本如果nvcc -V 命令运行后出来的是cuda11.3的话,其他版本请查看Previous PyTorch Versions | PyTorch
pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
2.导入torch出错
2.1.ImportError: cannot import name 'TypeAlias' from 'typing_extensions' (/usr/local/lib/python3.8/dist-packages/typing_extensions.py)
解决方案:
pip install typing-extensions --upgrade
3.运行程序出现的问题
3.1.TypeError: can't set attributes of built-in/extension type'torch._C.DisableTorchFunctionSubclass'
解决方案:
更换torch版本,出现以上错误大概率是电脑GPU/CUDA/CUDA toolkit不兼容。