cuda9.1环境安装torch报错

输入下面shell命令禁止使用cuda的头文件编译torch:

export TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__"

再次输入

 
 

./install.sh即可顺利安装。

猜你喜欢

转载自blog.csdn.net/longma666666/article/details/80760872