当前位置:   article > 正文

CUDA 12.2 配置pytorch简单版(Windows版本,GPU)_cuda 12.2 pytorch

cuda 12.2 pytorch

之前一直使用的虚拟机,但是等到跑深度学习模型的时候才发现,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

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

  1. conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  2. conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  3. conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  4. 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,便代表安装成功

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/煮酒与君饮/article/detail/972583
推荐阅读
相关标签
  

闽ICP备14008679号