This is just a project of ChatGPT3.5 demos.
-
install python 3.10+
-
update pip
python -m pip install --upgrade pip- install packages
python -m venv venv
. venv/bin/activate
pip install -r requirements.txt
python test01.py- run
python -m venv venv
python test01.py- 要匹配cuda版本及pytorch版本
# 查询cuda版本及pytorch版本
https://pytorch.org/get-started/locally/
https://developer.nvidia.com/cuda-toolkit-archive
# 下载所需cuda，比如118
# 安装对应版本的pytorch
# pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
# 测试一下
import torch
print(torch.cuda.is_available())