情况描述:
在复现别人代码的时候,发现该代码不仅使用了python还使用了c和c++语言,所以在运行的时候就出现了问题。
解决办法:
到setup.py所在位置比如:
cd ./codes/FAC/kernelconv2d/
运行:
python setup.py clean
如果出现:raise EnvironmentError('CUDA_HOME environment variable is not set. '
OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.
在命令行:
export CUDA_HOME=/usr/local/cuda-X.X
x.x是cuda版本,export不是内部命令则使用set替换,查看cuda版本
然后
python setup.py install --user
如果报错:
UserWarning: Attempted to use ninja as the BuildExtension backend but we could not find ninja.. Falling back to using the slow distutils backend.
安装该库
pip install ninja
如果报错:
C:\Users\test\Anaconda3\envs\cu100_py37_torch120\lib\site-packages\torch\utils\cpp_extension.py:189: UserWarning: Error checking compiler version for cl: [WinError 2] 系统找不到指定的文件。
则,将Visual Studio(没有就安装)中cl.exe
所在的目录添加到系统变量中的Path当中,
例如:C:\Program Files (x86)\Microsoft Visual Studio 14.0\VC\bin\amd64,当然如果你不知道到底是哪一个cl.exe就把四个都添加进去。
在cmd当中输入cl如果出现下面显示则安装成功
关掉命令行(或者cmd或者终端)重新打开输入命令
python setup.py install --user
有错误请指出!