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
    官网链接
    安装成功后,查看版本:
相关推荐
Francek Chen11 小时前
【深度学习计算机视觉】12:风格迁移
人工智能·pytorch·深度学习·计算机视觉·风格迁移
A-大程序员11 小时前
【Pytorch】分类问题交叉熵
人工智能·pytorch·分类
董建光d11 小时前
PyTorch 实现多种 CNN 模型并采用集成方法提升 CIFAR-10 分类性能
人工智能·pytorch·cnn
JJJJ_iii14 小时前
【机器学习03】学习率与特征工程、多项式回归、逻辑回归
人工智能·pytorch·笔记·学习·机器学习·回归·逻辑回归
woshihonghonga18 小时前
停止Conda开机自动运行方法
linux·人工智能·conda
liliangcsdn18 小时前
如何使用命令修改conda虚拟环境目录
conda
LiJieNiub1 天前
基于 PyTorch 实现 MNIST 手写数字识别
pytorch·深度学习·学习
chxin140161 天前
Transformer注意力机制——动手学深度学习10
pytorch·rnn·深度学习·transformer
MYX_3091 天前
第五章 神经网络的优化
pytorch·深度学习·神经网络·学习
久未1 天前
Pytorch autoload机制自动加载树外扩展(Autoload Device Extension)
人工智能·pytorch·python