基础知识
- pytorch在使用GPU的时候需要用到cuda toolkit,该工具是软件与硬件的桥梁。
- cuda有两个意思,一个是运行时cuda(cuda toolkit),一个是驱动器cuda(cuda driver)。通常我们的服务器上已经安装过了cuda driver,一般不需要再重新安装,所以本文所安装的cuda指的是安装运行时cuda,即cuda toolkit。
- 安装的cuda toolkit版本不能超过cuda driver的版本
查看cuda driver的版本
终端输入nvidia-smi
命令,下图的右上角即是服务器的cuda驱动版本(12.4)
在conda虚拟环境中安装cuda(cuda toolkit)
1) 进入想安装cuda的虚拟环境后,下载cuda toolkit安装包
2)根据自己系统的版本信息进行选择(我选的12.4):
<