【自用】在conda下安装不同的cuda版本(本次为cuda11.7)
运行时发现nvcc -V版本还是不合适(尝试在conda下安装nvcc)#安装cuda11.7和对应的pytorch等环境。#需要在cudatoolkit安装之后!#在python中查看cuda版本是否正确。#查看cudatoolkit的可安装版本。#查看包里面是否包含cuda 11.7。#安装cudatoolkit。#查看当前cuda情况。#查找对应版本nvcc。
·
参考博客:
[1]虚拟环境conda安装cuda11.7和cudnn和pytorch
[2]conda环境下安装nvcc -V
[3]解决 nvcc -V 显示的cuda版本不正确
#查看当前cuda情况
nvcc -V
#查看cudatoolkit的可安装版本
conda search cudatoolkit --info
#安装cudatoolkit
conda install cudatoolkit
#安装cuda11.7和对应的pytorch等环境
conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.7 -c pytorch -c nvidia
#查看包里面是否包含cuda 11.7
conda list
#需要在cudatoolkit安装之后!!!
conda install cudnn
#在python中查看cuda版本是否正确
python
>>>import torch
>>>print(torch.version.cuda) #11.7
>>>print(torch.backends.cudnn.version()) #8500
#运行时发现nvcc -V版本还是不合适(尝试在conda下安装nvcc)
#查找对应版本nvcc
conda search -c nvidia cuda-nvcc
#安装nvcc
conda install -c nvidia cuda-nvcc==12.2.140 #或者是11.7.99
欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐

所有评论(0)