50系列显卡配环境踩坑记录(pytorch

上个月入的5070Ti本来想美美跑代码了,但是发现很多package没有更新到适配cu12.8的版本(.........持续更新一个踩坑记录

1. 装torch的时候会报这个错:

NVIDIA GeForce RTX 5070 Ti with CUDA capability sm_120 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_50 sm_60 sm_70 sm_75 sm_80 sm_86 sm_90.

解决方式是只能装pytorch官网的适配cu128的nightly版本:

(哈哈 之前我确实是这样装成功的哈 但是4月14号配另一个环境的时候突然报错了说No matching distribution found for torch也不知道是怎么回事儿

2. 装xformers

xformers目前是完全没有提供cu128对应的包的好像,最后我按照github上贵人的指导进行了重新编译,可以参考:Fixed RTX 5090 build issue by maludwig · Pull Request #1235 · facebookresearch/xformers

### NVIDIA 50 Series GPU 上安装 PyTorch 的指南 为了在备 NVIDIA 50 系列显卡的设备上成功安装 PyTorch,需遵循以下方法和注意事项: #### 驱动程序准备 NVIDIA 显卡需要合适的驱动版本才能支持 CUDA 功能。对于最新的 NVIDIA GeForce RTX 50 系列显卡,建议先卸载可能存在的 Nouveau 开源驱动[^3],因为该驱动不提供完整的硬件加速功能。 可以按照以下命令禁用 Nouveau 并安装官方 NVIDIA 驱动: ```bash sudo apt-get remove --purge '^nvidia-.*' sudo add-apt-repository ppa:graphics-drivers/ppa sudo apt-get update sudo ubuntu-drivers autoinstall ``` 完成上述操作后重启计算机以加载新的 NVIDIA 驱动。 #### 安装 CUDA 和 cuDNN PyTorch 支持通过预编译包自动检测系统中的 CUDA 版本并下载对应的依赖项。然而,手动确认 CUDA 是否已正确置有助于减少潜在错误。可以从 NVIDIA 官方文档获取最新 CUDA 工具链的安装说明[^1]。 执行如下命令来验证 CUDA 安装状态: ```bash nvcc --version ``` 如果未发现任何可用的 CUDA 编译器,则应重新安装适合当前系统的 CUDA 软件包。 #### 下载与安装 PyTorch 针对特定硬件架构优化过的二进制文件能够显著提升性能表现。以下是基于 PythonPyTorch 安装方式之一,它会依据本地环境动态调整兼容性设置[^2]: ```bash pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 ``` 注意这里选择了 `cu118` 参数代表适CUDA 11.8 的构建版本;实际应用时可根据个人需求替换为其他标签(如无 GPU 加速可选用 cpu-only 类型)。 此外还有另一种途径利用 Conda 渠道实现更便捷的一键部署流程: ```bash conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch -c nvidia ``` 以上两种方案均能有效达成目标——即顺利集成 NVIDIA 50 系列 GPU 至 PyTorch 生态圈之中。 ---
评论 12
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值