配置pytorchgpu环境教程(较为简略,不懂的留言)

首先打开

然后

创建环境指令

 

然后

激活对应环境

conda activate torchyo

然后进入pytorch官网

然后

 

然后自己找对应版本下载

 复制相应指令 回车即可

安装完成后,打开pycharm

然后点开右下角 顺序点击,即可在编译器里配债好环境了。

 

 

最后打开脚本测试

import torch

print(torch.__version__)  # 输出pytorch版本信息

import torchvision

print (torchvision.__version__) # 输出torchvision的版本信息

print(torch.cuda.is_available()) # 输出True 说明CUDA可用
import torch
print(torch.cuda.is_available())
print(torch.backends.cudnn.is_available())
print(torch.cuda_version)
print(torch.backends.cudnn.version())

1.7.1
0.8.2
True
True
True
11.0
8004

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值