当前位置:   article > 正文

【自用】在conda下安装不同的cuda版本(本次为cuda11.7)

【自用】在conda下安装不同的cuda版本(本次为cuda11.7)

参考博客:
[1]虚拟环境conda安装cuda11.7和cudnn和pytorch
[2]conda环境下安装nvcc -V
[3]解决 nvcc -V 显示的cuda版本不正确

#查看当前cuda情况

nvcc -V
  • 1

#查看cudatoolkit的可安装版本

conda search cudatoolkit --info
  • 1

#安装cudatoolkit

conda install  cudatoolkit 
  • 1

#安装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
  • 1

#查看包里面是否包含cuda 11.7

conda list
  • 1

#需要在cudatoolkit安装之后!!!

conda install cudnn
  • 1

#在python中查看cuda版本是否正确

python
>>>import torch
>>>print(torch.version.cuda) #11.7
>>>print(torch.backends.cudnn.version()) #8500
  • 1
  • 2
  • 3
  • 4

#运行时发现nvcc -V版本还是不合适(尝试在conda下安装nvcc)
#查找对应版本nvcc

 conda search -c nvidia cuda-nvcc
  • 1

#安装nvcc

conda install -c nvidia cuda-nvcc==12.2.140 #或者是11.7.99
  • 1
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小桥流水78/article/detail/937062
推荐阅读
相关标签
  

闽ICP备14008679号