GPU: GeForce GTX 2080
- 安装pytorch1.4.0
默认安装的是cudatoolkit10.1,无法进行选择,故安装10.1
- 安装pytorch-encoding
pip install torch-encoding --pre
- 找不到detail,安装detail-api
repo_url = "https://github.com/zhanghang1989/detail-api"
os.system("git clone " + repo_url)
os.system("cd detail-api/PythonAPI/ && python setup.py install")
- 找不到Matplotlib ,安装!
- 使用默认参数 cuda out of memory!