1.环境配置
ubuntu20.04
RTX3090,显卡驱动nvidia 525(支持最高版本CUDA 12.0)---》sudo安装
python interpreter: python3.9 ----》python用sudo安装,在python interpreter中选择版本。
CUDA 11.6 GA与适配的CUDNN ----》sudo安装,在环境变量(全局bashrc或python中局部配置一下)中配置。
pytorch 1.13.1 torchvision==0.14.1
list如下:
$ pip list
Package Version
------------------- ------------
certifi 2023.11.17
charset-normalizer 3.3.2
contourpy 1.2.0
cycler 0.12.1
fonttools 4.46.0
idna 3.6
importlib-resources 6.1.1
joblib 1.3.2
kiwisolver 1.4.5
matplotlib 3.8.2
numpy 1.26.2
packaging 23.2
Pillow 10.1.0
pip 23.3.1
pyparsing 3.1.1
python-dateutil 2.8.2
requests 2.31.0
scikit-learn 1.3.2
scipy 1.11.4
setuptools 60.2.0
six 1.16.0
threadpoolctl 3.2.0
torch 1.13.1+cu116
torchaudio 0.13.1+cu116
torchsummary 1.5.1
torchvision 0.14.1+cu116
tqdm 4.66.1
typing_extensions 4.9.0
urllib3 2.1.0
wheel 0.37.1
zipp 3.17.0
DD-Net是在下面环境下写的:pytorch 1.3.0 torchvision==0.4.1
GitHub - BlurryLight/DD-Net-Pytorch: A lightweight network for body/hand action recognition.
上述配置完全可以跑通下面的DDnet,说明pytorch是向前兼容的。