1、CUDA安装
参考链接: cuda和cuDNN安装
2、GPU版本的torch安装
1.先用conda建立一个环境名称
2.去pytorch官网
参考链接: GPU版本的torch安装
3、jupyter notebook打不开问题
- 进入到新建的环境当中去
- 重新安装下jupyter即可在新的环境中打开jupyter notebook
pip install jupyter
4、对比
- 在cpu版本环境中打开jupyter notebook
检测不到显卡
- 在gpu版本的环境中打开jupyter notebook
5、总结
- 首先安装CUDAtoolkit,版本号最好低于自己电脑支持的CUDA
- 右键打开nvidia控制面板
- 左下角系统信息
- 点击组件
- 查看NVIDIA64.dll所支持的版本号
- 安装cuDNN,前面链接有介绍
- 用miniconda新建一个环境,进入环境
- 然后去pytorch找conda的下载链接,CUDA版本≤cudatoolkit版本号
- 安装d2l包,pip install d2l
- 如果在新的环境中打不开jupyter
- pip install jupyter