pytorch GPU版安装

pytorch GPU版安装

安装CUDA

• 查看本机已安装的CUDA版本的命令行指令：nvcc -V

• 查看本机显卡最高支持的CUDA版本的命令行指令：nvidia-smi

• 下载cuda 11.3：https://developer.nvidia.com/cuda-11.3.0-download-archive

• 解压

• 安装，这里要选自定义，精简会导致找不到安装目录

• 查看是否安装成功

安装pytorch

（使用这种方式可以避免出现错误）

• 下载地址https://download.pytorch.org/whl/torch_stable.html

• 下载完成后安装

• 测试，torch测试GPU命令print(torch.cuda.is_available())

安装完成！

torchvision安装

找到比当前pytorch版本高版本的torchvision，但是尽量选择接近的，比如我的torch版本torch-1.11.0+cu113-cp38-cp38-win_amd64.whl，那我就选择torchvision-0.12.0+cu113-cp38-cp38-win_amd64.whl。

下载链接https://download.pytorch.org/whl/torch_stable.html

pip install torchvision-0.12.0+cu113-cp38-cp38-win_amd64.whl