conda配置完整的pytorch虚拟环境

  1. 新建环境 conda create -n py38 python=3.8
  2. 虚拟环境中安装CUDA,conda安装的cudatoolkit和NVIDIA提供的CUDA Toolkit不一样,前者是系统CUDA的子集。在虚拟环境中安装了cudatoolkit,则pytorch就会用虚拟环境中的cudatoolkit进行编译。注意cudatoolkit版本要和NVIDIA Driver版本一致:
    NVIDIA官网:cudatoolkit和驱动版本对应
    安装cudatoolkit: 指定CUDA Toolkit版本安装
    根据上表查询到可安装的CUDA Toolkit版本,535.104.05对应最高的CUDA Toolkit版本为12.2,我安装12.1。
    运行conda install pytorch cudatoolkit=12.1 -c pytorch即可。
    此方法指定CUDA Toolkit版本后,conda会自动匹配到合适版本的PyTorch。
    如果提示找不到cuda版本:
    (1)换源
    (2)直接查询当前conda可安装版本:
bash 复制代码
conda search cudatoolkit --info
conda search cudnn --info

(3)pip方式安装

pip install -r requirements.txt

requirements.txt内容:

--extra-index-url https://pypi.ngc.nvidia.com

nvidia-cuda-runtime-cu12

nvidia-cuda-cupti-cu12

nvidia-cuda-nvcc-cu12

nvidia-nvml-dev-cu12

nvidia-cuda-nvrtc-cu12

nvidia-nvtx-cu12

nvidia-cuda-sanitizer-api-cu12

nvidia-cublas-cu12

nvidia-cufft-cu12

nvidia-curand-cu12

nvidia-cusolver-cu12

nvidia-cusparse-cu12

nvidia-npp-cu12

nvidia-nvjpeg-cu12

(4)官网安装,3种安装方式,我用的deb(local)
官网链接cuda12.1安装

安装成功后,查看版本:

  1. 安装cudnn
    官网链接
    安装成功后,查看版本:
相关推荐
TT-Kun4 小时前
PyTorch基础——张量计算
人工智能·pytorch·python
wydxry13 小时前
MOE架构详解:原理、应用与PyTorch实现
人工智能·pytorch·架构
MobiCetus13 小时前
确保conda环境内的Py不会污染系统
chrome·python·conda
天才少女爱迪生17 小时前
pytorch的自定义 CUDA 扩展怎么学习
人工智能·pytorch·学习
墨染枫1 天前
pytorch学习笔记-自定义卷积
pytorch·笔记·学习
冰镇西瓜1 天前
windows本地使用conda部署Open-webui
windows·ai·conda
SiYuanFeng2 天前
【问题未解决-寻求帮助】VS Code 中使用 Conda 环境,运行 Python 后 PowerShell 终端输出内容立即消失
开发语言·python·conda
柴 基2 天前
Visual Studio Code 使用指南 (2025年版)
人工智能·pytorch·python
码界筑梦坊2 天前
169-Django二手交易校园购物系统开发分享
后端·python·django·毕业设计·conda
z are2 天前
PyTorch 生态四件套:从图片、视频到文本、语音的“开箱即用”实践笔记
人工智能·pytorch·笔记·深度学习