Anaconda虚拟环境安装GPU版本的pytorch

1. 首先安装好Anaconda

2. 安装完之后Anaconda默认是没有桌面快捷方式的,需要点击【开始】,找到Anaconda,点击Anaconda Prompt

Anaconda虚拟环境安装GPU版本的pytorch

3. Anaconda配置清华镜像

由于Anaconda的conda install命令默认使用的是国外源下载,安装比较大的库(比如pytorch和cuda toolkit)的时候非常慢,所以建议配置Anaconda的下载源为国内镜像(比如清华,阿里或者豆瓣),本教程以清华镜像为例,在Anaconda Prompt 里输入如下指令:

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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

该指令的结果是在C:/user/username路径下产生一个名为.condarc的配置文件,只要不删除此文件即使卸载重装Anaconda下载源也不用重新设置。

4. 创建虚拟环境

在Anaconda prompt 里输入 _ conda create --name 虚拟环境的名字 python=3.7_

比如我想把虚拟环境叫做virtual_pytorch,则输入 _ conda create --name virtual_pytorch python=3.7_

(python = 3.7 是虚拟环境中的Python版本,可按需更改,目前用pytorch的话python装3.6或者3.7比较好)

这个虚拟环境就会安装在Anaconda安装目录下的envs文件夹里,如果不是安装在Anaconda下的envs而是安装在C盘下的某个文件夹里说明是权限不够,以管理员身份运行Anaconda Prompt就行。

5. 在虚拟环境里安装pytorch

在Anaconda prompt中输入activate envs_name激活虚拟环境,其中evns_name是你刚刚创建的虚拟环境的名字。前往pytorch官网获取下载指令。

Anaconda虚拟环境安装GPU版本的pytorch

复制红框中的指令,注意不要复制到了红框后面的 -c pytorch(它表示强制从默认的国外源安装pytorch)

cudatoolkit版本号要根据自己电脑的cuda版本号来修改(如果驱动有更新则先把显卡驱动升级到最新),可以在cmd里使用nvidia-smi命令查看自己电脑显卡的cuda版本,cuda version:11.1表示

支持的cuda版本小于等于11.1。

Anaconda虚拟环境安装GPU版本的pytorch

清华镜像里并不一定有每一个cuda版本的cudatoolkit包,可以自己去清华镜像看看:Index of /anaconda/cloud/pytorch/win-64/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror比如我的电脑cuda version:11.1,则我在pytorch官网获得的指令改为

conda install pytorch torchvision torchaudio cudatoolkit=11.1

我现在输进去之后报错显示没有cudatoolkit=11.1这个包,说明清华镜像里目前还没有在python3.7环境下的11.1版本的cudatoolkit,我换成11.0之后就有了。

conda install pytorch torchvision torchaudio cudatoolkit=11.0

6. 查看pytorch是否为gpu版本

进入python,输入

import torch
torch.cuda.is_available()

不用看了,结果是False,因为一般从清华镜像下载的默认是CPU版本,进入刚才创建的虚拟环境中使用conda list命令查看安装的包,有可能你会发现安装了一个叫cpuonly的包,pytorch果然也是cpu版的

Anaconda虚拟环境安装GPU版本的pytorch

这时候直接: conda uninstall cpuonly

他会提示你当你卸载cpuonly后一些函数库的版本改变(其中就包括pytorch 和 torchvision)

等待安装,loading…

安装完成后,就可以看到pytorch 和 torchvision 都是 GPU 版本了

此时输入

import torch
torch.cuda.is_available()

返回结果就是True啦

7. 可能没有cpuonly这个库怎么办

在输入 conda list查看包之后,发现库中并没有cpuonly包,但我的pytorch还是cpu版本的那么该怎么办呢?

解决方法: conda install cpuonly
你先自行给它装上cpuonly函数库
然后,你再卸载它: conda uninstall cpuonly

至此,gpu版本的pytorch就装好了

参考文章:https://blog.csdn.net/HuangJM3/article/details/123685177

Original: https://blog.csdn.net/m0_57110410/article/details/125529677
Author: 再游于北方知寒
Title: Anaconda虚拟环境安装GPU版本的pytorch

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/708505/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球