首页 > 其他分享 >GTX1050 安装GPU版pytorch流程

GTX1050 安装GPU版pytorch流程

时间:2022-10-29 17:25:47浏览次数:88  
标签:-- pytorch 虚拟环境 conda CUDA GTX1050 GPU 安装

  • 版本安装情况

Windows10+NVIDIA GTX 1050(笔记本版)+ Driver Version: 471.41 + CUDA 10.1 + python 3.7 + conda 4.10.1 + pytorch 1.7.1

  • Anaconda安装

官网进行下载:https://www.anaconda.com/

创建虚拟环境:conda create --name 环境名 python=版本号(我的执行为conda create --name pytorch python=3.7)

查看虚拟环境:conda info --envs

添加清华的镜像源:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

conda config --set show_channel_urls yes

  • CUDA下载

要找好自己显卡对应的CUDA版本,不建议下载最新的CUDA,因为可能与框架存在不匹配的问题。

本次下载的版本为CUDA10.1

接下来安装CUDA,选择自定义安装,我在E盘中建立了文件夹CUDA用于存放自定义安装时的内容(将其中一步中的前两个安装目录放在CUDA1,第三个安装目录放在CUDA2):

  • cuDNN安装

安装版本与CUDA10.1相对应,存在多个与CUDA10.1对应的cuDNN,选择其中一个非最新的,我选择的是cuDNN7.6.5

下载下来的为一个压缩包,将压缩包的内容复制到刚才创立的文件夹CUDA的CUDA1中(注意路径,粘贴到CUDA1中)

  • pytorch安装

使用pip的安装方法,进入到虚拟环境pytorch中进行安装(在cmd中使用指令conda activate pytorch),在pytorch中复制指令进行安装:

使用验证程序进行验证,发现安装成功:

 

标签:--,pytorch,虚拟环境,conda,CUDA,GTX1050,GPU,安装
From: https://www.cnblogs.com/gyymgirl/p/16241832.html

相关文章