3080 pytorch安装教程gpu

要在3080上安装GPU版本的PyTorch，需要先安装CUDA和相应的驱动程序。首先，确保你的系统上已经安装了NVIDIA显卡驱动程序，然后按照以下步骤进行操作：

1. 首先，根据你的操作系统下载并安装相应版本的CUDA。你可以在NVIDIA的官方网站上找到CUDA的下载链接和安装说明。

2. 下载并安装Anaconda，这是一个用于管理Python环境的工具。在安装过程中，请确保勾选添加Anaconda到系统环境变量。

3. 打开命令提示符或终端，并创建一个新的虚拟环境，可以使用以下命令：

   conda create -n pytorch_env python=3.8

4. 激活虚拟环境：

   conda activate pytorch_env

5. 安装PyTorch和torchvision。根据你的GPU型号和CUDA版本，选择合适的PyTorch版本。你可以在PyTorch的官方网站上找到版本对应关系。使用以下命令安装：

   conda install pytorch torchvision cudatoolkit=11.6 -c pytorch

6. 安装完成后，可以使用以下命令验证PyTorch和torchvision的版本：

   python -c "import torch; import torchvision; print('pytorch version:', torch.__version__); print('torchvision version:', torchvision.__version__)"

请注意，以上安装步骤假设你已经正确安装了NVIDIA显卡驱动程序和CUDA。如果在安装过程中遇到任何问题，请参考官方文档或在相关论坛上寻求帮助。

众所周知啊，在我们安装 GPU 版的 pytorch 之前，需要安装 CUDA，CUDA 的安装教程在此不过多追叙，详情可以看官方文档。
查看 torch 以及 torchvision 的版本
这里解释一下为什么会报错，原因是你 CUDA11.6 在官方网首页直接下载的 pytorch 三宝合集包里边，torchvision 这个是默认给你下 cpu 版本的，所以这边咱需要换一下它的版本。