一、安装pytorch
进官网,https://pytorch.org/get-started/locally/ ,找到自己想要的版本。
1、pytorch版本1.9.0,cuda版本1.0.2
pip3 install torch==1.9.0+cu102 torchvision==0.10.0+cu102 torchaudio===0.9.0 -f https://download.pytorc
h.org/whl/torch_stable.html
2、测试
import torch
x = torch.rand(5, 3)
print(x)
二、打包生成requirement.txt
# 安装
pip install pipreqs
# 在当前目录生成
pipreqs . --encoding=utf8 --force