项目的虚拟环境的搭建与pytorch依赖的下载

文章目录

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

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

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

配置环境

首先先创建虚拟环境

bash 复制代码
conda create --name <env_name> python=<python_version> 

激活虚拟环境

bash 复制代码
conda activate <env_name>

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

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

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

使用命令行检验

python 复制代码
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}")
bash 复制代码
PyTorch Version: 1.13.1
CUDA Available: True
CUDA Version: 11.7

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

bash 复制代码
# 但是我在使用下面的命令安装的时候,安装并不成功
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

出现下面的报错

bash 复制代码
  ERROR: Failed building wheel for torch-cluster
  Running 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)

参考资料:如何解决上面的问题?我们考虑直接在将库下载在本地,然后通过命令行进行安装

cpu 是支持cpu,否则cu 就是GPU

具体操作参考下面这个解决torch-cluster等无法下载的问题

解决torch-cluster等无法下载的问题
已解决python setup.py bdist_wheel did not run successfully.

除了对应的cuda和torch版本的问题,可能还存在其他问题
退出虚拟环境

bash 复制代码
conda deactivate
相关推荐
zzzyulin21 小时前
huggingface transformers调试问题--加载本地路径模型时pdb断点消失
python·transformer
教练、我想打篮球21 小时前
12 pyflink 的一个基础使用, 以及环境相关
python·flink·pyflink
飞翔的佩奇21 小时前
【完整源码+数据集+部署教程】【天线&运输】直升机战机类型识别目标检测系统源码&数据集全套:改进yolo11-CSP-EDLAN
前端·python·yolo·计算机视觉·数据集·yolo11·直升机战机类型识别目标检测系统
C嘎嘎嵌入式开发21 小时前
(21)100天python从入门到拿捏《XML 数据解析》
xml·开发语言·python
蓝博AI21 小时前
基于卷积神经网络的香蕉成熟度识别系统,resnet50,vgg16,resnet34【pytorch框架,python代码】
人工智能·pytorch·python·神经网络·cnn
小白银子1 天前
零基础从头教学Linux(Day 54)
linux·windows·python
不爱搬砖的码农1 天前
宝塔面板部署Django:使用Unix Socket套接字通信的完整教程(附核心配置与问题排查)
python·django·unix
滑水滑成滑头1 天前
**发散创新:探索零信任网络下的安全编程实践**随着信息技术的飞速发展,网络安全问题日益凸显。传统的网络安全防护方式已难以
java·网络·python·安全·web安全
丁浩6661 天前
Python机器学习---1.数据类型和算法:线性回归
开发语言·python·机器学习·线性回归
H_z_q24011 天前
Python动态类型、运算符、输入处理及算法编程问答
python