- 版本安装情况
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