conda环境安装cuda

cc嘿嘿嘿

于 2024-09-05 15:02:58 发布

阅读量208

点赞数 2

文章标签： linux 人工智能运维

本文链接：https://blog.csdn.net/ccheiheihei/article/details/141931392

版权

1、检查可以的cuda-nvcc版本

conda search -c nvidia cuda-nvcc

conda search cuda-nvcc

Loading channels: done
# Name                       Version           Build  Channel             
cuda-nvcc                    12.0.76     hba56722_10  /                   
cuda-nvcc                    12.0.76     hba56722_11  /                   
cuda-nvcc                    12.0.76     hba56722_12  /                   
cuda-nvcc                    12.0.76      hba56722_3  /                   
cuda-nvcc                    12.0.76      hba56722_4  /                   
cuda-nvcc                    12.0.76      hba56722_5  /                   
cuda-nvcc                    12.0.76      hba56722_6  /                   
cuda-nvcc                    12.0.76      hba56722_7  /                   
cuda-nvcc                    12.0.76      hba56722_8  /                   
cuda-nvcc                    12.0.76      hba56722_9  /                   
cuda-nvcc                   12.1.105      hba56722_0  /                   
cuda-nvcc                   12.1.105      hcdd1206_1  /                   
cuda-nvcc                   12.2.140      hcdd1206_0  /                   
cuda-nvcc                   12.3.107      hcdd1206_0  /                   
cuda-nvcc                   12.3.107      hcdd1206_1  /                   
cuda-nvcc                    12.4.99      hcdd1206_0  /                   
cuda-nvcc                   12.4.131      hcdd1206_0  /                   
cuda-nvcc                   12.4.131      hcdd1206_1  /                   
cuda-nvcc                    12.5.40      hcdd1206_0  /                   
cuda-nvcc                    12.5.82      hcdd1206_0  /                   
cuda-nvcc                    12.6.20      hcdd1206_0  /                   
cuda-nvcc                    12.6.68      hcdd1206_0  /

2、安装指定cuda-nvcc版本

conda install cuda-nvcc==12.3.107