3090显卡(CUDA11.1)安装Pytorch
由于官方暂时没有11.1版本的Pytorch安装方法,可以使用11.0版本的安装方法,经过测试,在3090上是可以正常使用的。
官方给出的安装方法:
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
在安装过程中,由于连接的是Pytorch的官方地址,所以下载的速度是很慢的,可以手动下载torch包和torchvison的包进行安装,运行一下上面的安装命令,就能看到地址。
另外,安装完torch包以后,在安装torchvison的包时,可能会自动下载另一个版本的torch包,可以通过
pip install --no-deps torchvision-0.8.2+cu110-cp38-cp38-linux_x86_64.whl
解决。
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 基于Microsoft.Extensions.AI核心库实现RAG应用
· Linux系列:如何用heaptrack跟踪.NET程序的非托管内存泄露
· 开发者必知的日志记录最佳实践
· SQL Server 2025 AI相关能力初探
· Linux系列:如何用 C#调用 C方法造成内存泄露
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 别再用vector<bool>了!Google高级工程师:这可能是STL最大的设计失误
· 单元测试从入门到精通
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 上周热点回顾(3.3-3.9)