RTX3060 cuda11.3一条命令安装pytorch1.10.0

最新推荐文章于 2024-06-22 14:57:37 发布

smallworldxyl

最新推荐文章于 2024-06-22 14:57:37 发布

阅读量2.9k

点赞数 5

文章标签： pytorch

本文链接：https://blog.csdn.net/smallworldxyl/article/details/121265442

版权

本文介绍了如何通过nvidia-smi命令检查NVIDIA GPU驱动版本，并强调了CUDA 11.3需要对应的驱动版本为465.89以上。接着，展示了创建和激活名为pytorch-gpu的conda环境的步骤，以及在这个环境中安装PyTorch 1.10.0+cu113及其相关库的过程。

摘要由CSDN通过智能技术生成

检查GPU驱动版本

在命令行输入nvidia-smi指令就能看到自己nvidia的驱动版本
在这里插入图片描述
NVIDIA官网，看下CUDA版本以及GPU驱动的对应关系：
可以看到要使用CUDA11.3，那么需要将显卡的驱动更新至465.89以上

1、pytorch-gpu环境的创建与激活

conda create –n pytorch-gpu python=3.7

activate pytorch-gpu

2.pytorch-gpu库的安装

activate pytorch-gpu

pip3 --default-timeout=60000 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio===0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple --trusted-host pypi.tuna.tsinghua.edu.cn