在深度学习中,CUDA被广泛应用于GPU加速。我们需要确保正确安装了所需要的CUDA版本。在Ubuntu或者CentOS上查看cuda,常用的方法是通过终端进入/usr/local目录下进行查看,但是这种方法比较繁琐,我们可以利用Anaconda来查看已安装的CUDA版本。
步骤:
$ source activate
$ conda install cudatoolkit
$ nvcc -V
下面是完整的代码示例:
$ source activate myenv
$ conda install cudatoolkit
$ nvcc -V
输出结果如下:
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Fri_Jun_25_19:15:27_PDT_2021
Cuda compilation tools, release 11.4, V11.4.48
Build cuda_11.4.r11.4/compiler.30033411_0
从结果中可以很容易地看到CUDA版本为11.4。
总结:
利用Anaconda查看已安装CUDA版本非常方便快捷。需要注意的是,安装cudatoolkit过程中可能会出现不兼容的问题,此时需要根据操作系统版本重新安装相应的CUDA版本。