用清华源安装pytorch(gpu)

安装pytorch-gpu打开清华源下的pytorch库
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
找到自己合适的版本,比如我需要GPU版本的并且支持cuda10,网页ctrl+f快捷搜索,找到文件中
-pytorch-1.2.0-py3.7_cuda100_cudnn7_1.tar.bz2版本号:1.2.0根据自己需求更改
nightly版本可以看作是开发版,一般不选用,命名格式为 pytorch-版本-python版本-cuda表示gpu版-cudnn版本.tar.bz2大小一般为5、600M下载后,放到合适的盘,接着在刚才创建的那个虚拟环境下,cd到文件目录
输入命令
conda install--offline pytorch-1.2.0-py3.7_cuda100_cudnn7_1.tar.bz2#文件全称安装完pytorch后,直接输入命令安装完依赖的包:conda install torchvision打开conda中的python进行测试
import torch print(torch.cuda.is_available())
print(torch.version.cuda)

  • 3
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值