终于在和环境的斗争中胜利了。
之前安装不上gpu版本大概是CUDA的原因,没有搞懂CUDA到底要装哪个版本,最主要的是到底要不要装。
大概碰到了以下几个问题。
- 显卡驱动是否安装
学长的判断方法,在桌面右键,看有没有NVIDIA控制面板。如果有,说明已经安装好了显卡驱动。没有的话,就需要去官网下载驱动安装了。
- 下载Anaconda
利用Anaconda安装python和pytorch等包,是非常方便的。
下载Anaconda可以到官网,建议到清华的镜像网站 https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/ 清华镜像
直接拉到最下面,下载最新的就好啦。
最下面的是python3.7版本的Anaconda.
- 安装pytorch-gpu
以往在这里踩坑最多。
打开Anaconda的Anaconda Prompt。(从开始那里就能找到)
首先,为了下载得快,或者下载不被中断,建议添加清华的镜像源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
接着,需要知道要不要下载CUDA!!!有些电脑是已经安装好CUDA的。
查看自己的CUDA信息:桌面空白处右键--->NVIDIA控制面板--->左下角有系统信息---->点击“组件”--->查看NVIDIA CUDA.DLL
于是我发现自己根本不需要下载安装CUDA23333,因为已经安装好了CUDA 10.1版本。
最后,只需要输入对应的10.1版本的Pytorch-gpu版本的命令:
conda install pytorch torchvision cudatoolkit=10.1
会下载很多包,选Y就完事了。
环境应该是装在base下的,没有自行创建环境。
- 验证是否安装成功
命令行输入python,敲以下的代码
import torch
import torchvision
print(torch.cuda.is_available())
返回为True则安装成功。