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

参考博客:
[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
  • 9
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值