文章目录
- 前言
- 一、conda环境
- 1.1 创建固定python版本的conda环境
- 1.2 激活 Conda 环境
- 1.3 关闭 Conda 环境
- 二、版本
- 查看CUDA版本
- 当电脑里有多个CUDN时
- 对照表
- 下载
- 资源
前言
一、conda环境
1.1 创建固定python版本的conda环境
conda create --name tang_py_3.12 python=3.12.4
1.2 激活 Conda 环境
conda activate <environment_name>
将 <environment_name> 替换为你要激活的环境的名称。例如,如果你想激活名为 myenv 的环境:
conda activate myenv
1.3 关闭 Conda 环境
conda deactivate
二、版本
查看CUDA版本
检查你的 GPU 驱动版本
nvidia-smi
验证 CUDA 版本
nvcc --version
当电脑里有多个CUDN时
博客:服务器-服务器里有多个CUDA
对照表
常见的Torch、CUDA和Python版本的对应关系
下载
- 使用 pip 安装 PyTorch 2.4(指定版本)并使用清华镜像源来加速下载。
慢
pip install torch==2.4 torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple
- pip 使用阿里云镜像源安装 PyTorch 2.4
快
pip install torch==2.4 torchvision torchaudio -i https://mirrors.aliyun.com/pypi/simple/
资源
阿里源