Pytorch-gpu安装教程
这是参考博文点击进入
下载想要的CUDA,原来下载的是CUDA11.2,但是发现pytorch不支持CUDA11.2.于是卸载掉CUDA11.2,如果不知道如何卸载请自行百度,重新下载CUDA11.3.高版本的CUDA支持向下兼容,举个例子也就是说算力7.0的显卡可以在支持最高算力7.5的CUDA版本下运行,但是算力7.5的显卡不可以在支持最高算力7.0的CUDA版本下运行,下载并安装好对应的CUDA和cudnn后就去网站下载好对应的pytorch的whl文件,网址在这点击进入,这是我选择下载的,我cu113是CUDA11.3,cp38是支持python3.8
下载到自己想要安装的环境的对应的Scripts文件夹下 ,然后打开cmd进入这个文件夹,激活这个环境,然后输入pip install whell
之后就可以进行安装了,输入pip install torch-1.10.0+cu113-cp38-
版权声明:
作者:lichengxin
链接:https://www.techfm.club/p/22051.html
来源:TechFM
文章版权归作者所有,未经允许请勿转载。
THE END
二维码
共有 0 条评论