pytorch 配置详细过程

时间:2023-01-27 16:05:06

pytorch 配置详细过程

github 项目多
方便,api好调用

cpu版本

装torch

安装最新版本的就可以。
pytorch 配置详细过程

torchvision

要版本对应
算法:
torchvision版本号=
torch版本号第一个数字-1.torch版本号第二个数字+1.torch版本号第三个数字

所以我的就是:

pip install torchvision==0.14.1 -i https://pypi.tuna.tsinghua.edu.cn/simple

成功效果:
pytorch 配置详细过程

GPU版本:

CUDA版本

先要确定CUDA:

nvidia-smi

pytorch 配置详细过程
这个是最高支持的版本:
pytorch 配置详细过程

然后我们去官网查看最新要求的版本:
pytorch 配置详细过程

可以看到是11.6

所以我们去网站上下载去11.6的
https://developer.nvidia.com/cuda-downloads

依次选择,然后下载就行
pytorch 配置详细过程

 nvcc -V

pytorch 配置详细过程
查看已安装的版本

https://developer.nvidia.com/cuda-toolkit-archive
下载历史版本
pytorch 配置详细过程

pytorch 配置详细过程
下载11.7版本

安装好之后,检查:
pytorch 配置详细过程

手动安装cuda的torch

http://download.pytorch.org/whl/torch/

下载对应版本的包。
pytorch 配置详细过程
torch version:
pytorch 配置详细过程

CPU的:
pytorch 配置详细过程

GPU的:
pytorch 配置详细过程
cu117就是cuda 11.7.x

python 版本:
pytorch 配置详细过程
cp38 就是 py 3.8.x

查看py版本:

pytorch 配置详细过程
我是3.9+

然后我是cuda11.7

windows平台

选定安装版本

pytorch 配置详细过程

pytorch 配置详细过程

开始安装:
pytorch 配置详细过程
成功

安装torchvision

同上:

http://download.pytorch.org/whl/torchvision/

pytorch 配置详细过程

pytorch 配置详细过程

成功:
pytorch 配置详细过程

测试GPU安装效果

import torch
print(torch.cuda.is_available())

pytorch 配置详细过程

返回True就是成功了

然后看一下 pip list
pytorch 配置详细过程
后面有这个加号就是可以的。

大功告成!!!!