1-1 配置pytorch+cuda环境

今日在跑代码过程中,出现如下错误:

Torch not compiled with CUDA enabled

首先,查看CUDA的版本,在cmd中输入

nvidia-smi

右上角 CUDA Version显示我的版本是11.0

我倾向于不改变CUDA的版本,但是主页并没有CUDA 11.0的选项(pytorch主页:PyTorch

点击下方的Previous Versions Of Pytorch,即https://pytorch.org/get-started/previous-versions/

此页面展示了不同版本pytorch+不同版本cuda的安装语句。

我的CUDA 11.0对应的版本是pytorch 1.7.1

如图,conda里的安装语句为(在conda中类似于cmd的页面运行,一次可能不成功,可多运行几次)

# CUDA 11.0
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch

如图,windows中的运行语句为(可在cmd中运行,一次可能不成功,多安装几次)

# CUDA 11.0
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

如果你的电脑中存在多版本的Python,安装语句则变为

cd C:\Users\(用户名)\AppData\Local\Programs\Python\Python38
python38 -m pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

这里使用python38是因为

我将C:\Users\(用户名)\AppData\Local\Programs\Python\Python38(下载后自带我未修改)中

下面的文件重命名为python38(我仅修改了该文件的名称)都配置好后再次运行代码,出现新的报错:

module 'torch' has no attribute 'nan'

查完资料发现tytorch1.8以上采用这个功能,好了我不得不去更新CUDA版本了,

安装CUDA,指路windows安装cuda11.1、cudnn及pytorch_cuda11.1对应的cudnn_烫烫烫烫烫烫丶的博客-CSDN博客我安装了CUDA11.7,并安装了最新版本的torch2.0.1,仍旧报错,

然后安装了torch1.13.1顺利解决问题,

安装哪个版本因程序而已,可以自己多试试

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值