第一次安装pytorch,记录一下,希望能帮助到有需要的人。
1、安装anaconda,官网教程安装_Anaconda 中文网
2、在pytorch安装中,最重要问题在于版本,和自己的显卡要对应。所以第一步是更新GPU(如果有的话)。更新网址在Official Drivers | NVIDIA。可以在设备管理器查看自己的GPU版本,选择对应的驱动下载
我的是1080
最后两项默认就可以(Download Type、Language)
安装好后在终端输入nvidia-smi可以查看自己的版本
右上角显示,比如我的是12.0
3、安装pytorch的方法很多,这里采用conda安装。在一些地方看到要安装cuda对应的版本,其实这一步是不用的,因为pytorch现在已经集成,直接进入pytorch官网就可以。PyTorch
第一行选择稳定版,第二行选操作系统,第三行选安装方式,第四行语言,第五行选版本。根据2的版本选择比他低的但是最靠近它的,我选择11.7,最后一行是对应的代码。
4、上述代码在哪里输入?
在开始菜单里打开anaconda prompt,出现
默认base环境,首先创建一个pytorch环境,依次输入下列代码
conda create -n pytorch python=3.6
conda activate pytorch
创建了pytorch环境,同时进入pytorch环境,出现
在此时输入第3步中的代码
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
即可进行安装。
5、遇到的问题
由于上述安装访问的是国外的网站,可能出现下载太慢的问题,很多文章里都提到了使用镜像的问题。比如,使用清华镜像,就是在第4步中进入pytorch之后,只要把最后一行的代码前依次加入两行:
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
然后最后一行去掉-c pytorch -c nvidia
但是,出现了问题,Collecting package metadata (current_repodata.json): failed,按照错误代码在网上进行了搜索,Collecting package metadata (current_repodata.json): failed(解决方案)_萝北村的枫子的博客-CSDN博客
用这篇文章的方法进行了修改,还是不行,发现是SSL不存在。CondaSSLError: OpenSSL appears to be unavailable on this machine.
在官网下载一个Win32/Win64 OpenSSL Installer for Windows - Shining Light Productions
我选择Win64OpenSSL_Light-1_1_1s,安装目录选择默认。安装好后,在anaconda prompt下继续安装又出现了下列问题
CondaSSLError: Encountered an SSL error. Most likely a certificate verification issue.
Exception: [SSL: DECRYPTION_FAILED_OR_BAD_RECORD_MAC] decryption failed or bad record mac (_ssl.c:2633)
是因为SSL没找到,最后发现将安装好的SSL放入anoaconda3/library/bin中就不会报错了。
最后,终于安装成功了
6、最后再验证一下
在pytorch环境下输出conda list,可以看到pytorch,当然还有torchvideo、 torchvision
最后输入python进入,调用一下torch输入,返回TURE即为成功,如果安装CPU版本就返回FALSE
torch.cuda.is_available()
其实,根据错误代码在网上找解决方法是最快最方便的,因为每个人的情况都不一样~