文章讲述了在安装GPU版PyTorch并尝试编译含CUDA算子的代码时遇到的问题。作者首先安装了cudatoolkit，但发现nvcc不在全局路径中。尝试通过conda和pip安装pytorch-cuda未果，后安装cuda11.1，并更新环境变量。在编译过程中遇到RuntimeError，通过安装ninja解决。最终成功运行代码。
NVCC
昨天已经安装好了gpu版的pytorch，对于一般的代码应该就可以运行了。但有些代码中需要用到cuda算子，需要配置nvcc环境。对于这个我也没能搞太清楚，网上的说法不一，我使用conda安装pytorch时也安装了cudatoolkit，按照我的理解，这里面应该带着nvcc，虽然没有在全局的path里面，但是在编译的时候应该可以找到这个路径。但事与愿违。
我在conda安装了cudatoolkit还是不能编译。看网上的教程，安装pytorch-cuda似乎可以，但是安装太慢，遂放弃。
我尝试的路径：
-
使用pip安装pytorch
不行，我的理解是没有安装cudatoolkit这个包
- 使用conda安装
这个方法应该是可以的，安装时包里有这个nvcc这个包，但是太慢了，遂放弃
-
尝试安装旧的版本，因为我复现论文的需要，我安装的是
conda install pytorch=1.9.0 torchvision cudatoolkit=11.1 -c pytorch -c nvidia -y，这里面也有cudatoolkit了，按照我的理解，应该是可以编译cuda算子的，但事与愿违 -
安装cuda，使用
apt install nvidia-cuda-toolkit，这时候有nvcc了，但是
最低0.47元/天 解锁文章
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
