python从零到pytorch【二】安装cuda和pytorch

1.下载地址

https://developer.nvidia.com/cuda-downloads

2.pytorch 选11.3   这的版本要小于等于第一步的版本     11.6需要conda-forge一直装不上 10.2不提供支持了

https://pytorch.org/

3.

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

4.按理来说 等一会就能完成安装

然后试一下是否安装成功

import torch
//查看torch版本 好像返回有问题
print(torch.version)
//查看cuda版本
print(torch.version.cuda)

//查看gpu是否可用
print(torch.cuda.is_available())

本文为看恩吧原创文章,转载无需和我联系,但请注明来自knsay.com

打赏

• 上一篇： python从零到pytorch【一】简单语法
• 下一篇： python从零到pytorch【三】类