一、系统配置
- windows10
- anaconda + pycharm(前提是已经安装的anaconda,后期需要用到其中的conda指令)
- gpu独显
二、涉及问题
- 如何使用conda安装pytorch, cudatoolkit等
- 解决pytorch等安装速度慢、卡顿的问题
- 如何查看自己的CUDA版本
- 如何选择合适的cudatoolkit
三、操作步骤
1. 确认自己电脑是否已经安装了NVIDIA驱动以及确认显卡驱动可以支持的最高CUDA版本(CUDA是英伟达公司为他们的显卡产品开发出来的一款机器学习框架)
dos中输入 nvidia-smi
得到如图所示,可以看出在输出中已经显示了cuda version为10.1,那这个是不是就是我们平常安装pytorch和xgboost-gpu版本所需要的cuda哪?从我理解的结果来说,这并不是我们所需要的,因为这个是NVIDIA驱动中自带的一个cuda driver api,是一个最底层的用于cuda计算的框架,但是基于其进行来开发比较复杂。我们一般使用的是cuda runtime api;
参考文献:关于cuda driver和cuda runtime的区别和联系
在nvidia-smi中显示的cuda版本和nvidia控制面板<NVIDIA控制面板-帮助-系统信息-组件-nvcud