Windows Pytorch 深度学习 环境配置
前言首先假设大家电脑拥有GPU步骤查看自己电脑GPU信息,打开终端输入指令nvidia-smi框起来的部分就是电脑能支持下载的最高CUDA版本- 接下来进入Pytorch官网查看各版本对应的信息。点击这里查看历史版本的对应信息如果大家执行官网上的命令存在问题的话,则进行这一步。根据自己电脑的CUDA版本下载对应的torch和torchversion至同一文件夹,文件夹位置没有要求。对于CUDA版本为10.1的伙伴们,按照下面的版本下载。进入torch的whl文件下载网站。torchvis
原创
2021-12-02 17:35:48 ·
258 阅读 ·
0 评论