安装pytorch纯小白教程||重点是要注意版本!

打开pytorch安装网址,根据系统配置选择特定版本

在Terminal输入命令,查看CUDA Version

nvidia-smi

 e.g.我的CUDA版本是11.2,下载CUDA11.8可向下兼容,因此对应我的操作系统这样选择

复制下面 Command 到 Terminal 中运行

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

 完成后运行以下代码进行测试

import torch
print(torch.__version__)
print(torch.cuda.is_available())

成功后最后一个会显示true

注意!

一开始其实是false,于是conda list了一下发现,pytorch-mutex是cpu版本,但是不管如何uninstall还是按照其他方法重下,重安再卸cpuonly,还是不行。索性把环境删了冲装,这次我装的python版本是3.9(上一次是3.7),于是奇迹般地True了!

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
您好!要安装GPU版本PyTorch,您可以按照以下步骤进行操作: 1. 首先,确保您的计算机上已经安装了适当的GPU驱动程序。您可以通过访问GPU制造商的官方网站来获取最新的驱动程序并按照说明进行安装。 2. 接下来,您需要安装CUDA(Compute Unified Device Architecture),这是一个用于在GPU上运行计算任务的并行计算平台。您可以从NVIDIA官方网站下载并安装适合您GPU的CUDA版本。 3. 安装CUDA后,您可以选择使用conda或pip来安装PyTorch。下面是两种方式的示例命令: 使用conda安装: ``` conda install pytorch torchvision torchaudio cudatoolkit=<CUDA版本号> -c pytorch ``` 使用pip安装: ``` pip install torch==<PyTorch版本号>+<CUDA版本号> torchvision torchaudio -f https://download.pytorch.org/whl/torch_stable.html ``` 请注意,将 `<CUDA版本号>` 替换为您在第二步中安装CUDA版本号,并将 `<PyTorch版本号>` 替换为您想要安装PyTorch版本号(例如:1.9.0)。 4. 安装完成后,您可以通过导入PyTorch并运行一些示例代码来验证是否成功安装了GPU版本PyTorch。例如: ```python import torch # 检查是否使用了GPU print(torch.cuda.is_available()) # 创建一个张量并将其移动到GPU上 x = torch.tensor([1, 2, 3]) x = x.to('cuda') # 输出张量 print(x) ``` 如果输出结果中显示了 `True`,则表示您已成功安装并配置了GPU版本PyTorch。 希望这些步骤对您有所帮助!如果您有任何其他问题,请随时提问。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值