当前位置:   article > 正文

Python环境配置详细说明,Pytorch,问题,详解_torchvision==0.3.0 python版本

torchvision==0.3.0 python版本

Python环境配置详细说明,Pytorch,问题,详解

具体版本配置还需要据根自身情况

我的是python3.7,cuda10.1 cudnn-7.6.5

https://www.pythonf.cn/read/135888

之前都是按照官网命令来安装,但是有时候官网这个下载速度实在是慢得让人受不了,这里给出另一种安装方式,用安装包来安装pytorch,利用清华镜像源和anaconda。

首先创建一个虚拟环境并激活

  1. conda create -n torch python=3.6
  2. conda activate torch

然后在清华源头下载安装包

https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

根据自己的需要,选择Linux系统还是win-64,进入之后可选取pytorch的版本以及cpu版还是gpu版

例如:我选择linux-64,然后选择python3.6、pytorch1.1.0、cuda10

同时还需要torchvision的话,可以选择python3.6、cuda10、torchvision0.3.0

安装包下好之后,选择好路径执行:

  1. conda install --offline pytorch-1.1.0-py3.6_cuda10.0.130_cudnn7.5.1_0.tar.bz2
  2. conda install --offline torchvision-0.3.0-py36_cu10.0.130_1.tar.bz2

这样就在torch的虚拟环境下装好了pytorch和torchvision,最后可能还需要自己安装cuda和cudnn,注意版本匹配

  1. conda install cudatoolkit=10.0
  2. conda install cudnn

这种方法就不用再担心网速的限制了,走清华源来下载速度非常快!

 

我的具体步骤:

测试是否安装成功:

 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/天景科技苑/article/detail/979913
推荐阅读
相关标签
  

闽ICP备14008679号