引用
安装CUDA
CUDA需独立安装
CUDA具体如何安装可以查看此博文:https://blog.csdn.net/GenuineMonster/article/details/116596761
Python 依赖安装
参考 依赖安装
遇到的问题
CUDA版本问题
首先安装CUDA11, 但是运行检测,无法检测到结果,卸载切换到CUDA10解决
pytorch版本问题
根据官网
pip install -r requirements.txt
会默认安装cpu版本,GPU版本需参照
https://blog.csdn.net/GenuineMonster/article/details/116604556
注意CUDA版本