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 小时前
包管理 pip ,conda;pycharm中使用conda 创建的虚拟环境
pycharm·conda·pip
N0nename4 小时前
TR3--Transformer之pytorch复现
人工智能·pytorch·python
MYX_3095 小时前
第七章 完整的模型训练
pytorch·python·深度学习·学习
CLubiy5 小时前
【研究生随笔】Pytorch中的线性代数
pytorch·python·深度学习·线性代数·机器学习
学不会就看5 小时前
PyTorch 张量学习
人工智能·pytorch·学习
兰文彬5 小时前
Pytorch环境安装指南与建议
人工智能·pytorch·python
淮北49420 小时前
windows安装minicoda
windows·python·conda
StarPrayers.1 天前
卷积神经网络(CNN)入门实践及Sequential 容器封装
人工智能·pytorch·神经网络·cnn
新子y1 天前
【小白笔记】PyTorch 和 Python 基础的这些问题
pytorch·笔记·python
我是李武涯1 天前
PyTorch DataLoader 高级用法
人工智能·pytorch·python