【踩坑记录】关于pytorch安装(有关cuda安装)

        最近因为跑transformer模型,要配置环境。torch_cpu版本比较好处理,直接在anaconda上输入对应的命令行就可以自动下载。比较经典的事torch1.8.1,在很多教程和案例中都使用到了这个torch。

不经常使用N卡,所以驱动、cuda、cudnn也要重新安装。

        安装驱动没什么讲究,直接通过网址选择对应的驱动下载安装即可:NVIDIA GeForce 驱动程序 - N 卡驱动 | NVIDIA

        顺便附上torch的下载地址,里面包含了多种torch_cuda版本:download.pytorch.org/whl/torch_stable.html

下载cuda要和torch_cuda的版本号对应,比如torch_cuda11.1要想在对应的11.1CUDA:

CUDA Toolkit Archive | NVIDIA Developer

重点:cuda安装的时候,exe文件解压的位置不要和后面软件安装的位置一致,不然软件安装完成后,解压文件夹的内容会别删除。这也是我疑惑了很久的问题,为什么每次安装完成都没有。

安装CUDA完成之前需要配置系统变量,变量是什么安装过程有提示的。

最后是cudnn的安装,其实就是个压缩包解压,把对应文件夹的内容放到cuda里面:cuDNN Archive | NVIDIA Developer

  • 9
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值