安装pytorch——GPU版本环境

本文的主要是叙述怎么去安装安装pytorch——GPU版本的,
背景是我装了很久的pytorch gpu版本但是最后都用不了GPU
看土堆老师的视频https://www.bilibili.com/video/BV1S5411X7FY?p=30&vd_source=5d20c2da06d505f27c78d07b9447a08e

觉得有收获,决定把笔记分享出来,希望对你会有一点点帮助

首先要创建环境,我试的版本是python = 3.6

命令 conda create --name yourEnv python=3.6,我觉得应该是没所谓,3.7、3.8、3.9应该都可以

然后,这里面会有一个坑!!!

创建环境完成后,不要着急安装pytorch!!!

去pip list看一下自己的环境中是不是已经装了pytorch 的cpu版本!!!

如果有,一定要先卸载掉!!!

我发现的来由是我的电脑存储空间很小,照常理创建一个虚拟环境大小不会超过1GB,但是我创建环境后一看消失了两个2GB

有问题:只要装过pytorch的人都知道pytorch大概是1.2GB,1+1.2=2.2,然后检查一看果然直接安装了CPU版本的

导致我后面不管怎么装都使用不了GPU

所以,创建环境后,先看一下有没有pytoch的cpu版本有的话,卸载掉

conda uninstall pytorch

如果有空间不够的小伙伴,下面这个命令应该用得上

(38条消息) conda清理没用的安装包_ypfzhao的博客-CSDN博客_conda清理无用包

conda clean -p //删除没有用的包
conda clean -t //tar打包
conda clean -y -all //删除所有的安装包及cache
这个命令,在你卸掉一些东西后,它的相关东西用不到可以卸掉,这时会比较有用

然后土导的视频:最详细的 Windows 下 PyTorch 入门深度学习环境安装与配置 CPU GPU 版 | 土堆教程

会有点长,觉得慢那就从P21开始看,
如果你是一个小白,你就从P1开始慢慢看,它会系统的给你解释很多东西,你会有收获的

安装命令

这里为什么不用官网的命令呢?当然是慢了

大家都知道访问国外的东西都很慢,或者直接不行

-c是channel的意思,后面的链接是镜像,镜像就是把国外的资源放在国内服务器上了,所以快咯

1.conda install cudatoolkit=11.3 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main

2.2.conda install pytorch torchvision torchaudio -c https://mirror.nju.edu.cn/pub/anaconda/cloud/pytorch/win-64/

验证:
import torch
print(torch.cuda.is_available())
希望你会是True

本文的技术干货很少,但是这里面的土堆视频,很有用
大家在有些问题反复出现的时候,可能得思考一下是不是开始就错了

你可能感兴趣的:(pytorch,python,深度学习)