CUDA+torch+vscode安装教程(python3.8)

编程知识 更新时间:2023-04-06 18:43:00

方法一: 

pytorch官网:Previous PyTorch Versions | PyTorch

推荐)我自己安装了一下环境,感觉这个教程不错:配置stylegan2环境遇到的一些问题 - 知乎

pip install torch==1.8.2+cu111 torchvision==0.9.2+cu111 torchaudio===0.8.2 -f https://download.pytorch/whl/lts/1.8/torch_lts.html -i  https://pypi.tuna.tsinghua.edu/simple

或者安装可以匹配CUDA11的最低版本1.7.1,这样可能能适配一些老项目

pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch/whl/torch_stable.html


CUDA的cuDNN在我的U盘上,当然你也可以在这个链接下载:

链接
cudnn-11.1-windows-x64-v8.0.4.30
链接:https://pan.baidu/s/1aqedvmiMXANVQqV5NPm7Yw
提取码:3ccj

cuda_11.1.0_456.43_win10
链接:https://pan.baidu/s/1KnhrYxptLQohVQ5Atql4sA
提取码:1td5
 

方法二 


首先通过命令:安装python3.8的环境

conda create -n torch38 python=3.8

我的解决方法如下:

这里一定要注意python版本和cuda要对应,否则会错误

1、https://download.pytorch/whl/torch_stable.html

找到这个:cu111/torch-1.8.0%2Bcu111-cp38-cp38-win_amd64.whl

 2、然后查看torch和torchvision对应关系要对应:

参考下面连接:

https://github/pytorch/vision#installation

3、这是下载完毕后的文件


在终端 pip install … (文件名).whl(这里要注意在该文件夹下安装,并且切换到你要安装的conda环境)

先安装torch

pip install "torch-1.8.0+cu111-cp38-cp38-win_amd64.whl" -i http://pypi.douban/simple --trusted-host pypi.douban

然后安装torchvision


即可安装完毕

这里我把这两个下载好的文件放在了U盘里,以后就可以直接安装了

 Pytorch gpu(CUDA11.1)最新版本 官网路径 安装超时_氟小汀的博客-CSDN博客

更多推荐

CUDA+torch+vscode安装教程(python3.8)

本文发布于:2023-04-06 18:43:00,感谢您对本站的认可!
本文链接:https://www.elefans.com/category/jswz/5409a236a794c7c871ddb4be9133e3b1.html
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。
本文标签:教程   torch   CUDA   vscode

发布评论

评论列表 (有 0 条评论)
草根站长

>www.elefans.com

编程频道|电子爱好者 - 技术资讯及电子产品介绍!

  • 50371文章数
  • 14阅读数
  • 0评论数