之前一直使用的虚拟机,但是等到跑深度学习模型的时候才发现,cuda和pytorch不匹配,报错如下:AssertionError: Torch not compiled with CUDA enabled。并且更致命的事情是,虚拟机无法调用主机中的GPU,故直接下载Windows版本的。

电脑的显卡是4060的,cuda版本是12.2,查看的方法可以从我之前的博文获取。但是pytorch官网(PyTorch)目前只更新到cuda12.1的。

 为此,查了好多方法,其中有一个方法就是将cuda降级,但是博主觉得实在是太麻烦了。便直接下载了最新的pytorch,意外的发现可以匹配,并且完全可以用。故分享方法如下:

首先,下载Anaconda,并且按照步骤进行安装即可(选择的文件夹需要为空文件夹,不然会报错)。安装完之后,在开始的菜单里面,点击下图圈出来的,打开类似于linux系统的终端:

 打开之后,创建环境:

conda create -n pytorch python=3.8

pytorch是我自己定义的名字,可以根据自己的选择进行更改。

之后输入

conda activate torch

出现如下图所示的便为激活成功:

然后便可以去官网下载pytorch,建议下载stable版本的。

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

由于网站是国外的原因,下载速度是很感人的,因此可以通过以下命令添加镜像源,下载速度会快一些(比不添加快一点只是,不知道怎么回事,和虚拟机上面的速度没法比)

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/pytorch/

conda config --set show_channel_urls yes

然后再把上面的pytorch下载链接改为如下的:

conda install pytorch torchvision torchaudio pytorch-cuda=12.1

在solving environment如果提示比如:EnvironmentNotWritableError: The current user does not have write permissions to the target environm。一般是没有足够的权限,解决方法是,在打开上面的Anaconda Powershell Prompt的时候,选择以管理员身份运行即可。之后问题便被成功解决,最后,看cuda和pytorch是否匹配:

首先python,然后输入;

import torch

然后,

torch.cuda.is_available()

如果返回True,便代表安装成功

相关文章

评论可见,请评论后查看内容,谢谢!!!
 您阅读本篇文章共花了: