Anaconda中的cudatoolkit是什么

Anaconda中的cudatoolkit是什么。

在使用anaconda安装配置pytorch环境的时候,常常需要安装cudatoolkit。这里的cudatoolkit与CUDA Toolkit

NVIDIA官网上下载安装的CUDA环境,包含了nvidia驱动程序、CUDA开发环境(包括NVCC编译器、调试器、相关的头文件等)、CUDA文档、CUDA示例等一系列资源。

而Anaconda提供的cudatoolkit是一系列编译好的动态库文件。将pytorch常用的各种cuda函数编译好提供出来,只要显卡驱动版本适合,就可以直接调用其中的API。

通常情况下,只需要安装cudatoolkit就可以正常使用。但是如果需要编写Pytorch的CUDA/C++ Extension,即手动通过CUDA实现新的功能(Layer/Operator)等,就必须保证机器上安装了官方的CUDA环境,从而实现动态的编译。

Lei Yang
Lei Yang
PhD candidate

My research interests include visual speech recognition and semantics segmentation.