当前位置：首页 > news >正文

国外移民公司网站设计网站建设苏州

news 2025/7/18 14:17:31

国外移民公司网站设计,网站建设苏州,网页设计实训报告300字,广州海珠做网站文章目录配置环境 pytorch的使用需要安装对应的cuda 在PyTorch中使用CUDA, pytorch与cuda不同版本对应安装指南，查看CUDA版本，安装对应版本pytorch 【超详细教程】2024最新Pytorch安装教程（同时讲解安装CPU和GPU版本） 配置环境…

文章目录

配置环境

pytorch的使用需要安装对应的cuda

在PyTorch中使用CUDA, pytorch与cuda不同版本对应安装指南，查看CUDA版本，安装对应版本pytorch

【超详细教程】2024最新Pytorch安装教程（同时讲解安装CPU和GPU版本）

配置环境

首先先创建虚拟环境

conda create --name <env_name> python=<python_version>

激活虚拟环境

conda activate <env_name>

下载对应版本的pytorch

并不建议不查看对应的pytorch与你的cuda版本是否匹配就直接下载，这样会导致后续出现不适配的问题

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

在这个网站，找到对应版本的下载的命令，注意要与你的cuda适配

使用命令行检验

import torch# 检查版本和CUDA可用性
print(f"PyTorch Version: {torch.__version__}")
print(f"CUDA Available: {torch.cuda.is_available()}")
if torch.cuda.is_available():print(f"CUDA Version: {torch.version.cuda}")

PyTorch Version: 1.13.1
CUDA Available: True
CUDA Version: 11.7

Pytorch也有相关的拓展库，当需要安装的时候，可以采用多种方式安装

# 但是我在使用下面的命令安装的时候，安装并不成功
pip install torch-cluster -f https://data.pyg.org/whl/torch-1.10.1+cu117.html
pip install torch-scatter==2.0.9 -f https://data.pyg.org/whl/torch-1.10.1+cu117.html
pip install torch-sparse==0.6.12 -f https://data.pyg.org/whl/torch-1.10.1+cu117.html
pip install torch-geometric==2.0.3

出现下面的报错

  ERROR: Failed building wheel for torch-clusterRunning setup.py clean for torch-cluster
Failed to build torch-cluster
ERROR: ERROR: Failed to build installable wheels for some pyproject.toml based projects (torch-cluster)