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

後面有這個加號就是可以的。

大功告成！！！！