1. 创建虚拟环境：

conda create -n python310_test python=3.10

2. 激活虚拟环境

conda activate python310_test

3. 安装pytorch （好像只能是pip安装，不能是conda安装）（这一步可以参考安装pytorch）

pip install torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 --index-url https://download.pytorch.org/whl/cu121

4. 设置虚拟环境中的环境变量

 conda env config vars set CUDA_PATH="/usr/local/cuda-12.1/targets/x86_64-linux"

这里的/usr/local/cuda-12.1/targets/x86_64-linux是路径，得看你的 cuda.h 和 nvrtc.h 文件在哪里，我这里的这两个文件在/usr/local/cuda-12.1/targets/x86_64-linux/include，那么就用它的上一级目录，有些服务器上的这两个文件在/home/XXX/Anoconda3/envs/myproject/include，那么路径就用/home/XXX/Anoconda3/envs/myproject

5. 查看是否设置成功

echo $CUDA_PATH

如果输出你刚刚的路径，则设置成功

6. 安装pykeops

pip install pykeops

搞了两天，终于用上面的流程装上了这个包