Windows 根据github上的环境需求,安装一个虚拟环境,安装cuda和torch

比如我们在github上看到一个关于运行环境的需求

Installation

  • xxx系统
  • Python 3.xxx
  • CUDA 9.2
  • PyTorch 1.9.0
  • xxx
  • xxx

最主要的就是cuda和torch,这两个会卡很多环境的安装。

我们重新走一遍环境安装。

首先创建一个虚拟环境

conda create -n 环境名字 python=3.xxx

activate 环境名字

设置一个全局定义,清华源

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

一般的库文件 xxx可以用国内源的形式非常快速下载

比如

pip install  -i https://pypi.tuna.tsinghua.edu.cn/simple tensorboardX

pip install  -i https://pypi.tuna.tsinghua.edu.cn/simple xxx

其次安装cuda和torch的版本,需要根据这个需求来,我看到需求是

  • CUDA 9.2
  • PyTorch 1.9.0

建议看下自己cuda的版本

打开命令行cmd
输入

nvidia-smi

可以看到自己cuda版本是12.2,但是并没有luan用。根据12.2这个版本,找到相近的cuda 11.1也能用

根据代码需求,我们在网站找到合适版本的代码。

网站如下:Previous PyTorch Versions | PyTorch

# CUDA 11.1
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

但是"https://download.pytorch.org/whl/torch_stable.html"通过这个网站下载速度非常非常慢

这时候我们需要通过链接外网形式,不用国内源

pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html     (需要梯子)

就这样完成对所有所需库文件的安装!!!

相关推荐
是十一月末25 分钟前
Opencv实现图片的边界填充和阈值处理
人工智能·python·opencv·计算机视觉
机智的叉烧1 小时前
前沿重器[57] | sigir24:大模型推荐系统的文本ID对齐学习
人工智能·学习·机器学习
凳子花❀1 小时前
强化学习与深度学习以及相关芯片之间的区别
人工智能·深度学习·神经网络·ai·强化学习
量子-Alex2 小时前
【多模态聚类】用于无标记视频自监督学习的多模态聚类网络
学习·音视频·聚类
吉大一菜鸡2 小时前
FPGA学习(基于小梅哥Xilinx FPGA)学习笔记
笔记·学习·fpga开发
泰迪智能科技013 小时前
高校深度学习视觉应用平台产品介绍
人工智能·深度学习
盛派网络小助手3 小时前
微信 SDK 更新 Sample,NCF 文档和模板更新,更多更新日志,欢迎解锁
开发语言·人工智能·后端·架构·c#
Eric.Lee20214 小时前
Paddle OCR 中英文检测识别 - python 实现
人工智能·opencv·计算机视觉·ocr检测
cd_farsight4 小时前
nlp初学者怎么入门?需要学习哪些?
人工智能·自然语言处理
AI明说4 小时前
评估大语言模型在药物基因组学问答任务中的表现:PGxQA
人工智能·语言模型·自然语言处理·数智药师·数智药学