环境搭建——CUDA、Python、Pytorch

安装anaconda

创建虚拟环境

conda create -n env_name python==python版本

安装pytorch

在官网中下载:

https://pytorch.org/get-started/previous-versions/

https://pytorch.org/get-started/locally/

测试cuda是否可用

python 复制代码
import torch

# 打印设备信息
print("Available devices:")
print(torch.device('cpu'))
print(torch.device('cuda'))
print(torch.device('cuda:1'))

# 检查 CUDA 是否可用
is_cuda_available = torch.cuda.is_available()
print(f"CUDA is available: {is_cuda_available}")

# 获取可用的 GPU 数量
gpu_count = torch.cuda.device_count()
print(f"Number of GPUs available: {gpu_count}")

# 打印每个 GPU 的名称
if is_cuda_available:
    for i in range(gpu_count):
        print(f"GPU {i}: {torch.cuda.get_device_name(i)}")
else:
    print("No CUDA-capable GPU is available.")

:::success
Available devices:

cpu

cuda

cuda:1

CUDA is available: True

Number of GPUs available: 2

GPU 0: NVIDIA GeForce RTX 3090

GPU 1: NVIDIA GeForce RTX 3090

:::

在测试cuda可用之后再去安装依赖库

:::color1

pip install -r requirements.txt

:::

对应关系

相关推荐
潮湿的心情6 分钟前
亚洲牧原:活跃行业交流,延伸公益版图,市场拓展再结硕果
大数据·人工智能
平和男人杨争争7 分钟前
机器学习14——线性回归
人工智能·机器学习·线性回归
嘉琪00114 分钟前
2025 js——面试题(7)——ajax相关
开发语言·javascript·ajax
SoniaChen3316 分钟前
Rust基础-part3-函数
开发语言·后端·rust
一个天蝎座 白勺 程序猿19 分钟前
飞算JavaAI进阶:重塑Java开发范式的AI革命
java·开发语言·人工智能
liu_yueyang21 分钟前
JavaScript VMP (Virtual Machine Protection) 分析与调试
开发语言·javascript·ecmascript
全干engineer22 分钟前
Flask 入门教程:用 Python 快速搭建你的第一个 Web 应用
后端·python·flask·web
前端 贾公子22 分钟前
tailwindCSS === 使用插件自动类名排序
java·开发语言
10岁的博客23 分钟前
代码编程:一场思维与创造力的革命
开发语言·算法
七七七七0723 分钟前
C++类对象多态基础语法【超详细】
开发语言·c++