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
    官网链接
    安装成功后,查看版本:
相关推荐
阳懿1 小时前
【已解决】执行conda init提示No action taken.
conda
try2find1 小时前
移动conda虚拟环境的安装目录
linux·运维·conda
喝过期的拉菲15 小时前
如何使用 Pytorch Lightning 启用早停机制
pytorch·lightning·早停机制
kk爱闹15 小时前
【挑战14天学完python和pytorch】- day01
android·pytorch·python
Yo_Becky20 小时前
【PyTorch】PyTorch预训练模型缓存位置迁移,也可拓展应用于其他文件的迁移
人工智能·pytorch·经验分享·笔记·python·程序人生·其他
xinxiangwangzhi_20 小时前
pytorch底层原理学习--PyTorch 架构梳理
人工智能·pytorch·架构
FF-Studio21 小时前
【硬核数学 · LLM篇】3.1 Transformer之心:自注意力机制的线性代数解构《从零构建机器学习、深度学习到LLM的数学认知》
人工智能·pytorch·深度学习·线性代数·机器学习·数学建模·transformer
盼小辉丶1 天前
PyTorch实战(14)——条件生成对抗网络(conditional GAN,cGAN)
人工智能·pytorch·生成对抗网络
Gyoku Mint1 天前
深度学习×第4卷:Pytorch实战——她第一次用张量去拟合你的轨迹
人工智能·pytorch·python·深度学习·神经网络·算法·聚类
郭庆汝1 天前
pytorch、torchvision与python版本对应关系
人工智能·pytorch·python