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