背景
Pytorch作为现在深度学习领域主流的框架之一,恰逢毕设中模型使用Pytorch进行搭建,赶上现在不想做毕设,就来记载下之前我电脑中的Torch1.7虚拟环境配置,啊哈哈哈
补充Tensorflow_GPU安装:https://blog.csdn.net/qq_46343832/article/details/121542099
环境说明
- Windows10_64位
- Torch版本:version1.7
- cuda10.1(原因:安装的为GPU版本,具体选择Cuda可以查看我前面文章Tensorflow中有说明)
具体安装步骤
- 使用Conda创建存放Torch的虚拟环境,并进入:
# 删除虚拟环境
conda remove -n your_env_name(虚拟环境名称) --all
# 创建虚拟环境,采用3.8环境
conda create -n torch1.7 python=3.8
# 查看虚拟环境命令
conda env list
# 进入虚拟环境命令
conda activate torch1.7
2. 进入Pytorch官网,找到Pytorch1.7的GPU版本,注意如果你电脑比较牛,显卡驱动支持,也只直接装最新的版本,步骤都一样:
3. 找到Tourch1.7对应的Conda命令复制到cmd中的虚拟环境中进行安装,可能安装过程比较缓慢,如果你有时间等下去,可直接等它安装好就可使用,如果中间部分第三方库丢失,可重新执行该命令。如果你不想等,请继续往下看,执行另一种安装方式。
4. 如果你也出想像我这情况,肯定不想等待吧,哈哈哈哈哈!!!那下面我们开始另外一种安装方式
① 使用conda安装下cudatoolkit=10.1,cuda选择只要比你电脑驱动支持的cuda版本低就没的问题(查看自己电脑的cuda版本支持看文章开头里说的Tensorflow_GPU安装里有说明)
conda install cudatoolkit=10.1
如果安装还是比较慢!建议换下安装源,比如清华源等
② 手动下载torch1.7.1的压缩包.whl文件,再进行pip的安装,比如,我们找到官网提供的torch1.7的pip安装如下:
pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
然后我们其后半部分https://download.pytorch.org/whl/torch_stable.html进入官方存放torch压缩文件的位置
③ 搜索torch-1.7.1%2Bcu101
找到https://download.pytorch.org/whl/cu101/torch-1.7.1%2Bcu101-cp38-cp38-win_amd64.whl
,鼠标右击复制链接地址,采用迅雷下载,为啥用迅雷呢?懂的都懂!!!!
④ cmd进入下载好的文件所在目录中
⑤ 使用命令pip install "torch-1.7.1+cu101-cp38-cp38-win_amd64.whl" -i https://pypi.douban.com/simple/
进行安装
⑥ 对pip安装中剩余的第三方库进行安装
pip install torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple/