环境介绍
- python3.9.6
- win11
- cuda12.0 -> cuda11.8
cuda原来安装成12.0了,发现没有与之匹配的torch。对其进行卸载后,重新安装cuda11.8版本
学习路线
学习沐神的动手学深度学习
关于cuda/gpu/cpu 以及python之间的关系,推荐观看小土堆的视频讲得很详细
环境安装
cuda的卸载与安装
- 卸载
win+R输入control,查找cuda,卸载以下四个软件
我这里是已经卸载重装过的,所以是11.8,原来为12.0 - 安装
访问地址下载自己想要的版本,下载完成后打开,全部默认就行了
# 查看cuda版本
nvcc -v
torch安装
在pycharm终端中使用pip进行安装,你可以使用virtualenv库去新建自己的虚拟环境,也可以在pycharm右下角添加解释器
安装之前你需要查看torch与cuda的版本适配
# 在终端运行安装torch相关
pip3 install torch torchvision torchaudio -i http://pypi.doubanio.com/simple/ --trusted-host pypi.doubanio.com
# 安装jupyter
pip install jupyter
# 打开jupyter
jupyter notebook
沐神的d2l在这个网址,下载后解压放在项目文件夹里面,然后使用jupyter打开就可以进行愉快的学习了