域名出售网站,2023年最新科技新闻摘抄,卸载ghost版wordpress,企业网站html源代码文章目录 配置环境 pytorch的使用需要安装对应的cuda 在PyTorch中使用CUDA, pytorch与cuda不同版本对应安装指南#xff0c;查看CUDA版本#xff0c;安装对应版本pytorch
【超详细教程】2024最新Pytorch安装教程#xff08;同时讲解安装CPU和GPU版本#xff09;
配置环境… 文章目录 配置环境 pytorch的使用需要安装对应的cuda 在PyTorch中使用CUDA, pytorch与cuda不同版本对应安装指南查看CUDA版本安装对应版本pytorch
【超详细教程】2024最新Pytorch安装教程同时讲解安装CPU和GPU版本
配置环境 首先先创建虚拟环境 conda create --name env_name pythonpython_version 激活虚拟环境 conda activate env_name下载对应版本的pytorch 并不建议不查看对应的pytorch与你的cuda版本是否匹配就直接下载这样会导致后续出现不适配的问题 https://pytorch.org/get-started/previous-versions/
在这个网站找到对应版本的下载的命令注意要与你的cuda适配 使用命令行检验 import torch# 检查版本和CUDA可用性
print(fPyTorch Version: {torch.__version__})
print(fCUDA Available: {torch.cuda.is_available()})
if torch.cuda.is_available():print(fCUDA Version: {torch.version.cuda})PyTorch Version: 1.13.1
CUDA Available: True
CUDA Version: 11.7Pytorch也有相关的拓展库当需要安装的时候可以采用多种方式安装 # 但是我在使用下面的命令安装的时候安装并不成功
pip install torch-cluster -f https://data.pyg.org/whl/torch-1.10.1cu117.html
pip install torch-scatter2.0.9 -f https://data.pyg.org/whl/torch-1.10.1cu117.html
pip install torch-sparse0.6.12 -f https://data.pyg.org/whl/torch-1.10.1cu117.html
pip install torch-geometric2.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)参考资料如何解决上面的问题我们考虑直接在将库下载在本地然后通过命令行进行安装 cpu 是支持cpu,否则cu 就是GPU 具体操作参考下面这个解决torch-cluster等无法下载的问题 解决torch-cluster等无法下载的问题 已解决python setup.py bdist_wheel did not run successfully. 除了对应的cuda和torch版本的问题可能还存在其他问题 退出虚拟环境 conda deactivate
