第一次在CSDN发文章,不是教程,只是备忘录!
1)cuda、cudnn的卸载
用" windows + I " 快捷键打开设置,找到应用卸载,在搜索栏输入" NVIDIA ",查阅前人经验,除去下图红色框三个,其余全部卸载,但是我在卸载时全都卸载了,后面也还是成功安装了,并且成功运行。
2) cuda、cudnn的重装
为了能够同时使用pytorch和tensorflow的gpu版本,需要安装两者都能用的cuda版本,这里选择的是cuda11.0和cudnn8.0,参考的链接如下:
同时安装 Tensorflow&Pytorch.
cuda11.0下载链接:link.
cuda、cudnn详细安装步骤:link.
3)pytorch安装
打开anaconda prompt,创建一个新的环境,
conda create -n 环境名 python=3.7
输入 y 以进行下一步,
接着激活创建好的环境:
activate 环境名
本人安装的是pytorch 1.7.1版本,cu110表示cuda版本为11.0,激活环境后输入安装代码:
pip install torch