1、顯卡驅(qū)動531.41，cuda 12.1，cuda里包含驅(qū)動與geforce ex...軟件與physX。

2、裝cuda tookit還需要vs desktop c++的相關(guān)支持。

3、Pytorch 2.0可以去官網(wǎng)查一下代碼下載安裝：

========= PyTorch + Xformers =========

PyTorch2.0 + cu118 + Xformers0017 + cuDNN

++++++++++++++++++++++++++++++++++

CUDA-Tookit:

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

PyTorch:

https://pytorch.org/get-started/locally

VS2022：

https://visualstudio.microsoft.com/zh-hans/downloads/選擇第一個community(個人免費版)

//-------------------------------------------------------//

(穩(wěn)定版Pytorch 2.0)

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

(測試版Pytorch)

pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118

pip install xformers==0.0.17