PyTorch笔记 安装

python版本:3.7
CUDA版本:10.0

安装相应的PyTorch的cmd命令:

conda install pytorch torchvision cuda100 -c pytorch

安装速度还是挺快的(不到10分钟)

注意:安装前以管理员身份运行cmd

测试程序:

import torch

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

a = torch.randn(2,3)
print(a)

运行结果:
在这里插入图片描述

更新:
再次安装时,使用原来的方法遇到了问题,新的解决方法如下:
python版本:3.7
CUDA版本:10.0
INSTALLING PREVIOUS VERSIONS OF PYTORCH 搜索“CUDA 10.0”:
在这里插入图片描述
可以发现,需要安装PyTorch 1.2.0,但使用网站给的安装命令:

conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

出现了问题

解决办法:
https://download.pytorch.org/whl/cu100/torch_stable.html
下载torch-1.2.0-cp37-cp37m-win_amd64.whl
在这里插入图片描述
下载完后在下载到的路径下运行:

pip install torch-1.2.0-cp37-cp37m-win_amd64.whl

即安装完成。

安装torchvision同理

相关文章:
INSTALLING PREVIOUS VERSIONS OF PYTORCH
How to Install PyTorch with CUDA 10.0

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值