pytorch 环境安装

文章讲述了如何通过nvidia-smi查看CUDA版本,强调了CUDA的向下兼容性。内容包括如何在不同平台上安装PyTorch(CPU/GPU,pip和conda方式),以及TensorFlowGPU的安装步骤,特别提到了离线安装的注意事项。还介绍了在PyCharm中集成tensorflow的详细教程。
摘要由CSDN通过智能技术生成

 nvidia -smi 显示的cuda版本是当前驱动最高支持的版本

nvcc -V现在的是自己的cuda版本

cuda 是向下兼容的,比如pytorch需要11.4,但是我们的是11.6是可以下载这个pytorch版本使用的

标蓝的其实就是cuda Toolkit的二进制文件

Python与PyTorch的版本对应_python和pytorch对应版本-CSDN博客

https://blog.csdn.net/shiwanghualuo/article/details/122860521

对应关系

安装教程:ubuntu安装pytorch的CPU版本GPU版本,分conda安装和pip安装,以及whl离线安装pytorch_conda离线安装whl-CSDN博客

1:使用pip

1.1在线直接复制pytorch官网OSX,下的命令,换成清华源配置

1.2离线安装(pip 换源之后速度较快一般用不到)

pytorch环境搭建(离线方式) - 哔哩哔哩

2:使用conda

2.1在线直接复制命令

如果有下不了的包考虑离线下载,离线下载步骤将下载不了包的网站复制到浏览器命令行下载

2.2不推荐conda离线下载,原因pytorch会给你新的渠道优先级高于导致离线下载的包,使得离线下载的包没用

3tensorflow gpu下载

环境cuda

https://blog.csdn.net/weixin_41012765/article/details/124973351

在里面找有的版本

本地cudaTensorflow-gpu-2.7.0安装教程和接入PyCharm(学生党详细教程,win10,Anaconda3,python3.9)_pycharm安装tensorflow-CSDN博客

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值