Pytorch 安装 GPU版本

2 篇文章 0 订阅
2 篇文章 0 订阅

1. 查看自己的CUDA版本,打开cmd:

nvcc --version

 所以CUDA 版本为10.2。 

 2. 打开 pytorch官网 ,选择适合自己的安装命令:

以管理员身份运行!以管理员身份运行!Anaconda Prompt !(不然,可能报错)

(即使报错,只要你不断地重新输入命令,直到它可以安装为止!因为有可能是网络问题。)

切到自己所要安装pytorch的环境,输入以上Command:

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

当然,也可以下载以前的Pytorch版本:https://pytorch.org/get-started/previous-versions/ 

 ————————耐心等待下载————————

3. 验证是否下载成功:

  1. 输入python
  2. 输入 import torch
  3. 输入 torch.cuda.is_available() 查看是否能调用gpu,没有报错即安装成功。


参考链接:

pytorch安装最容易实现版本,踩坑无数总结

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
安装PyTorchGPU版本,首先需要确保你的电脑具备支持CUDA的显卡。根据引用中提到的,你的电脑使用的是GeForce MX150显卡,它是支持CUDA的。 接下来,需要按照以下步骤安装PyTorchGPU版本: 1. 首先,下载并安装适用于你的CUDA版本CUDA Toolkit。CUDA Toolkit是一个用于支持GPU计算的软件开发工具包。你可以从NVIDIA官方网站上下载与你的CUDA版本相对应的Toolkit。 2. 安装CUDA Toolkit后,打开Anaconda Prompt,输入以下命令来创建一个新的虚拟环境(可选步骤): ``` conda create --name myenv ``` 其中,"myenv"是你给虚拟环境起的名字,你可以根据自己的需求进行命名。 3. 激活刚才创建的虚拟环境: ``` conda activate myenv ``` 4. 接下来,使用conda命令来安装PyTorchGPU版本。根据引用中提到的,选择适合你的PyTorch和torchvision版本。例如,如果你想要安装PyTorch的1.4.0版本和torchvision的0.5.0版本,可以使用以下命令: ``` conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=<your_cuda_version> -c pytorch ``` 其中,"<your_cuda_version>"是你所安装CUDA Toolkit的版本号。 5. 等待安装完成后,就成功安装PyTorchGPU版本。 总结一下安装PyTorchGPU版本的步骤: 1. 确保你的电脑具备支持CUDA的显卡。 2. 下载并安装适用于你的CUDA版本CUDA Toolkit。 3. (可选步骤)创建一个新的虚拟环境。 4. 激活虚拟环境。 5. 使用conda命令安装PyTorchGPU版本。 希望这些步骤对你有帮助!

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值