Pytorch 在anaconda环境安装时的问题及解决(CUDA11.2)

文章描述了一位初学者在Anaconda环境下遇到CUDA11.2与PyTorch版本不匹配的问题。尝试安装PyTorch1.10.0与cudatoolkit11.3后失败,发现可能是由于Python版本过高。降低Python版本至3.8后,成功解决了安装问题,并证实新环境能支持GPU(3060)运行PyTorch。

由于是初学者,文章主要目的是记录学习过程,可能不是非常完备。


这里是在anaconda中建立了“CNN_E”的新环境,问题出在之前安装的CUDA版本为11.2。

但离谱的是pytorch官网没有对应版本。

在别的文章的建议下,尝试下载pytorch==1.10.0 对应cudatookit=11.3。(指令如下)

(图1)

但是运行不成功,报错如下图

(图2)

问题可能出现在环境中python的版本,不兼容or ...?(感觉有时候很玄学)

总之看报错意思可能是python版本有点高。于是新建了一个环境,测试是不是猜想的问题。(如下图)

(图3)

将python版本降到3.8。

(图4)

依然运行‘图1’安装程序,成功解决问题。

(图5)

经检验也能够使用电脑GPU。(ps:演示电脑gpu为3060,不知道gpu不一样有没有影响)

(图6)

=============================END==============================

安装支持 CUDA 11.2PyTorch 库,并且在 Python 3.8 环境下运行,可以使用 PyTorch 官方提供的安装命令,也可以通过 Anaconda 进行安装。以下是具体方法: ### 使用 pip 安装 可以通过指定 PyTorch 的版本和 CUDA 版本来安装适合的 PyTorch 包。例如,使用以下命令: ```bash pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html ``` 此命令中,`torch` 和 `torchvision` 的版本均支持 CUDA 11.1,通常也兼容 CUDA 11.2。需要注意的是,PyTorch 版本和 CUDA 版本之间需要匹配,否则可能导致安装失败或运行错误。 ### 使用 Anaconda 安装 如果使用 Anaconda,则可以通过以下命令安装支持 CUDA 11.2PyTorch: ```bash conda install pytorch torchvision torchaudio cudatoolkit=11.2 -c pytorch ``` 此命令会自动安装CUDA 11.2 兼容的 PyTorch 和相关库。此外,确保已经配置了正确的 Conda 渠道,例如清华源,以提高下载速度: ```bash 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 是否成功识别了 CUDA: ```python import torch print(torch.__version__) print(torch.cuda.is_available()) print(torch.version.cuda) ``` 如果输出显示 `torch.cuda.is_available()` 为 `True`,并且 `torch.version.cuda` 返回 `11.2`,则表示安装成功,并且 PyTorch 已经正确配置了 CUDA 支持 [^4]。 ---
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

End!ess

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值