11.14 杂项配置gpu torch 环境

python@king

已于 2022-11-15 14:48:47 修改

阅读量293

点赞数

分类专栏：盈透金融人工智能文章标签： python 开发语言

于 2022-11-14 12:21:57 首次发布

本文链接：https://blog.csdn.net/yueguoqing88/article/details/127845104

版权

在尝试安装torch-1.4.0+cpu-cp38-cp38m-win32.whl时遇到错误，通过修改安装包名称为torch-1.4.0+cpu-cp38-none-win32.whl并使用pip安装，问题得到解决。如果仍无法成功，可以运行`python -m pip debug --verbose`进行排查。最后，检查torch的CUDA可用性。

摘要由CSDN通过智能技术生成

import setuptools.pep425tags   #根据自己pep425tags文件位置进行相应更改
print(setuptools.pep425tags.get_supported())

报错：ERROR: torch-1.4.0+cpu-cp38-cp38m-win32.whl is not a supported wheel on this platform.
解决方法：
在自己的python文件夹的site-packages中寻找pep425tags.py文件

2、修改为 pip install torch-1.4.0+cpu-cp38-none-win32.whl，安装成功

3、上面的都不好使用这行代码 python -m pip debug --verbose

4、torch.cuda.is_available()

import os
import torch
from torch import nn
from torch.utils.data import DataLoader
from torchvision import datasets, transforms

device = "cuda" if torch.cuda.is_available() else "cpu"
print(f"Using {device} device")

class NeuralNetwork(nn.Module):
    def __init__(self):
        sup