【新手】 nvidia-smi:11.6 nvcc-V:11.3不匹配 Python:3.9下安装Torch:1.10.2、Torchvision:0.11.3)

1.nvidia-smi和nvcc-V版本不一致

nvidia-smi

显示版本为11.6
在这里插入图片描述

nvcc -V

显示版本为11.3
在这里插入图片描述
参考:https://blog.csdn.net/sophicchen/article/details/120782209
nvidia-smi版本大于nvcc -V的版本不会有什么问题,并且安装python包(例如torch)需要匹配runtime cuda版本,即nvcc -V的版本。

2. 创建虚拟环境conda命令

创建python版本为3.9的虚拟环境 conda create -n '环境名' python==3.9

激活虚拟环境conda activate '环境名'
在当前环境中查看anaconda安装路径conda info --envs

3. 官网查看版本

在官网查看对应的版本https://pytorch.org/

pip3 install torch==1.10.2+cu113 torchvision==0.11.3+cu113 torchaudio===0.10.2+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html 

但是太慢!

4.下载pytorch-1.10.2

推荐离线下载:https://download.pytorch.org/whl/torch_stable.html
下载自己版本对应的 .whl 文件:

  • 根据cuda版本(11.3)+torch(1.10.2)+python版本(3.9)+windows系统——选择版本进行下载
  • cu113 + torch-1.10.2 + cp39 + win + 64
    在这里插入图片描述

4.下载torchvision-0.11.3

下载自己版本对应的 .whl 文件:

  • 根据cuda版本(11.3)+torchvision(0.11.3)+python版本(3.9)+windows系统——选择版本进行下载
  • cu113 + torch-1.10.2 + cp39 + win + 64
    在这里插入图片描述
    下载的文件:
    在这里插入图片描述

5. 安装pytorch-1.10.2 并验证是否成功

在当前虚拟环境下,使用conda命令安装下载的pytorch-1.10.2
安装pytorch-1.10.2pip install + whl文件位置
下图显示是因为我已经安装过了
在这里插入图片描述
查看当前虚拟环境安装的包 命令:conda list

验证是否安装成功:
代码

import torch
print(torch.__version__)  #注意是双下划线
print(torch.version.cuda)
print(torch.cuda.is_available())
print(torch.cuda.get_device_name())

打印:输出为true,说明已经正确安装。

在这里插入图片描述

6. 安装 torchvision-0.11.3

在当前虚拟环境下,使用conda命令安装下载的 torchvision-0.11.3
安装torchvision-0.11.3pip install + whl文件位置
在这里插入图片描述
查看当前虚拟环境安装的包 命令:conda list
在这里插入图片描述

  • 11
    点赞
  • 21
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值