GTX1060+win10+CUDA11.3+cudnn8.2+pytorch1.11.0——个人配置踩坑日记

以下是我亲测有效的使用 GTX 1060 的各部分安装版本
电脑系统:window 10
python版本:3.8.13
pytorch版本:1.11.0
CUDA版本:11.3
cuDNN版本:8.2

以下是我个人安装过程的步骤,以及我踩过的坑,这里解决踩坑问题主要参考:

RTX3060+win10+CUDA11.2+cudnn8.2.0+pytorch1.8.0 环境——个人配置经验
Win10+MX250+CUDA11.3+cuDNN+Pytorch安装

文章目录

  • 前言
  • 一、安装Miniconda
  • 二、创建conda环境
  • 三.安装pytorch深度学习框架和d2l软件包
  • 四、pytorch-gpu安装
  • 五、测试

前言

由于本人今年CS准研一小白,入门深度学习的过程中,在学长的推荐下学习了李沫大神的《动手学深度学习》,在安装的pytorch环境的过程钟遇到的一些坑,在此特意分享踩坑心得,也防止下次换电脑重新装新的环境的时候避免再次遇到同样的问题。


一、安装Miniconda

这里根据李沫大神的《动手学深度学习》的安装教程进行安装的,大家也可以选择安装完整的conda。最简单的方法就是安装依赖Python 3.x的Miniconda。 如果已安装conda,则可以跳过以下步骤。

1、进入Miniconda官 :https://conda.io/en/latest/miniconda.html

conda create –name torch_gpu python=3.8

conda activate torch_gpu

我们可以使用conda list命令来查看当前conda环境中有哪些包。

如果是老的显卡驱动可能cuda版本是10.2,这里我提前更新了一下显卡驱动,因为pytorch官 说win10不支持10.2cuda,支持11.3。

nvcc -V

如图所示则为成功

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

默认pytorch官 为下载源,下载速度太慢,很容易 错,所以更改为清华大学镜像,命令行输入下面的命令:

#添加Anaconda的清华镜像
conda config –add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config –add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config –add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
#设置搜索时显示通道地址
config –set show_channel_urls yes

conda uninstall pytorch

注:如果要删除自定义源更换回conda的默认源,直接删除channels即可,命令如下:

conda config –remove-key channels

然后采用官 默认的源进行安装就成功了,就是速度很慢。

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

五、测试

最后安装完成之后再进行验证。

通过nvcc -Vnvidia-smi分别查看cuda版本


声明:本站部分文章及图片源自用户投稿,如本站任何资料有侵权请您尽早请联系jinwei@zod.com.cn进行处理,非常感谢!

上一篇 2022年4月3日
下一篇 2022年4月3日

相关推荐