Win10+RTX3080Ti显卡安装Pytorch记录

Win10+RTX3080Ti显卡安装Pytorch记录

显卡驱动安装

**方法一：**下载鲁大师安装所需驱动
**方法二：**下载驱动精灵安装所需驱动
**方法二：**进入NVIDIA官网手动搜索驱动程序，然后默认安装

安装Visual C++支持（要使用CUDA,必须要有一个编译工具）

进入Visual Studio官网，选择社区版Visual Studio 安装。（我安装的是VS2017，安装选择里面的Visual C++组件即可，其他按需安装）

安装CUDA和CUDNN

Win+R–>运行页面输入cmd输入nvidia-smi命令查看当前驱动所需的CUDA。去CUDA官网下载，默认安装，（不重新安装驱动，我安装的11.5）

默认安装后，环境变量中自动有CUDA_PATH, 若没有手动新建。
Win+R–>运行页面输入cmd，输入nvcc -V后回车有CUDA相关信息，即安装成功。

去CUDNN官网下载对应版本的CUDNN

将下载下来的文件解压复制到安装路径中

安装Pytorch

默认已经安装好anaconda和配置好国内源。

#配置虚拟环境
conda create -n py37 python=3.7
#创建完成后进入环境
conda activate py37
#安装pytorch
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
#pip超时，加上timeout的时间
pip install --default-timeout=1000 torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

为了测试pytorch是否成功安装，在环境下输入python，输入以下代码，返回True则表示安装成功。

import torch
torch.cuda.is_available()

pytorch下载链接

从官网下载需要的软件包版本