Can‘t find libdevice directory ${CUDA_DIR}/nvvm/libdevice

构建的乐趣2023-11-25 20:57

win10 Running deepxde 的时候出现问题：

cuda-nvcc 安装后解决了。

cpp 复制代码

# Install NVCC
conda install -c nvidia cuda-nvcc=11.3.58 -y
# Configure the XLA cuda directory
mkdir -p $CONDA_PREFIX/etc/conda/activate.d
printf 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/\nexport XLA_FLAGS=--xla_gpu_cuda_data_dir=$CONDA_PREFIX/lib/\n' > $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
source $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
# Copy libdevice file to the required path
mkdir -p $CONDA_PREFIX/lib/nvvm/libdevice
cp $CONDA_PREFIX/lib/libdevice.10.bc $CONDA_PREFIX/lib/nvvm/libdevice/
这是一段linux命令，我想在win10上做类似的事情，因为我遇到了Can't find libdevice directory ${CUDA_DIR}/nvvm/libdevice 报错以及ptxas returned an error during compilation of ptx to sass 报错

我只用了第一个命令，安装nvcc库就完事了