pytorch 配置详细过程

2024-10-21 05:39:28

torch

github 项目多

方便，api好调用

cpu版本

装torch

安装最新版本的就可以。

torchvision

要版本对应

算法：

torchvision版本号=

torch版本号第一个数字-1.torch版本号第二个数字+1.torch版本号第三个数字

所以我的就是：

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

成功效果：

GPU版本：

CUDA版本

先要确定CUDA:

nvidia-smi

这个是最高支持的版本：

然后我们去官网查看最新要求的版本：

可以看到是11.6

所以我们去网站上下载去11.6的

https://developer.nvidia.com/cuda-downloads

依次选择，然后下载就行

 nvcc -V

查看已安装的版本

https://developer.nvidia.com/cuda-toolkit-archive

下载历史版本

下载11.7版本

安装好之后，检查：

手动安装cuda的torch

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

下载对应版本的包。

torch version:

CPU的：

GPU的：

cu117就是cuda 11.7.x

python 版本：

cp38 就是 py 3.8.x

查看py版本：

我是3.9+

然后我是cuda11.7

windows平台

选定安装版本

开始安装：

成功

安装torchvision

同上：

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

成功：

测试GPU安装效果

import torch

print(torch.cuda.is_available())

返回True就是成功了

然后看一下 pip list

后面有这个加号就是可以的。

大功告成！！！！

最新文章

热门文章