三种方式
1 nvcc
2usr local
3torch
图像增强库
albumentations
import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())
import torchvision
torchvision.__version__
后台运行:
nohup
torch版本对应关系
https://blog.csdn.net/shiwanghualuo/article/details/122860521
各版本下载网址
https://download.pytorch.org/whl/torch_stable.html
torch torchvision torchaudio
1.11.0 0.12.3 0.11.0
pip install torch==1.6.0+cu101 -i https://mirrors.aliyun.com/pypi/simple/