win10 Running deepxde 的时候出现问题:
cuda-nvcc 安装后解决了。
# 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库就完事了
后来又出现这个 报错以及INTERNAL: libdevice not found at ./libdevice.10.bc,我发现是自己CUDA cudnn根本没装。CUDA按报错提示装:
2023-11-28 11:23:11.538320: W tensorflow/compiler/xla/service/gpu/nvptx_helper.cc:56] Can't find libdevice directory ${CUDA_DIR}/nvvm/libdevice. This may result in
compilation or runtime failures, if the program we try to run uses routines from libdevice.
Searched for CUDA in the following directories:
./cuda_sdk_lib
C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.2
/usr/local/cuda
说明我应该装11.2 cuda 以及cudnn 装好重启一下即可