conda install cudatoolkit之后,在环境中找不到CUDA_HOME, export之后(其实export也没用,根本就没安装嘛),又找不到nvcc,单独安装nvcc后又报找不到cuda_runtime_api.h
无语的cuda安装,怎么这么多事儿。
后来查到解决方法是
conda install -c conda-forge cudatoolkit-dev
Cudatoolkit Dev :: Anaconda.org
然后就都解决了。
ps 安装时Executing transaction会比较久,耐心等待。注意版本不要搞太高,最好和cudatoolkit一致吧。
找了半天,且用且珍惜。