操作系统 windows 11 专业版

CUDA Version: 12.1

python 3.12.7

显卡信息:RTX 4060 Ti

安装 torch 2.3.1

conda install pytorch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 pytorch-cuda=12.1 -c pytorch -c nvidia

验证


在学习pytorch过程中可能会有多个项目同时进行,不同项目之间使用的 Python 版本和一些库的版本不一样,这就会导致冲突。因此建议使用 Anaconda 来管理多个 Python 虚拟环境和pytorch。
Anaconda 是为了方便使用 Python 而建立的一个软件包,包含常用 的 250 多个工具包,多个版本的 Python 解释器和强大的虚拟环境管理工具。各个环境之间相互独立,可任意切换。

安装前需要准备什么

情况一:已装过Python的小伙伴

请删掉自己电脑的本来的python,因为之后可能会干扰到安装。或自行搜索“Python安装Pytorch教程”相关内容。


情况二:未装过Python的小伙伴

1. 确定自己所需安装的pytorch是GPU还是CPU版本。

2.下载anaconda并pytorch其配置环境

3.通过anaconda下载安装pytorch

4.下载pycharm,并为程序配置环境

简单三句话
前往官网安装anaconda。
Anaconda换源。
在conda终端中使用虚拟环境安装pytorch环境

conda环境使用python虚拟环境常见命令
1、查看服务器存在的虚拟环境
conda env list 
或 conda info -e 
2、创建虚拟环境
conda create -n env_name python=X.X
3、激活虚拟环境
conda activate env_name
4、安装依赖包
pip install -r requirments.txt    (ps:先来到FAME-master路径,不然会找不到requirments文件)
5、关闭虚拟环境 
conda deactivate
6、运行代码
python main.py
7、删除虚拟环境
conda remove -n your_env_name(虚拟环境名称) --all

conda环境使用python虚拟环境常见命令
Windows:

1、文件管理器文件路径地址栏敲:%APPDATA% 回车,快速进入 C:\Users\电脑用户\AppData\Roaming 文件夹中
2、新建 pip 文件夹并在文件夹中新建 pip.ini 配置文件
3、我们需要在pip.ini 配置文件内容,你可以选择使用记事本打开,输入以下内容,输入完后记得按下ctrl+s保存哦,在这里我们使用的是豆瓣源
[global]
index-url = http://pypi.douban.com/simple
[install]
use-mirrors =true
mirrors =http://pypi.douban.com/simple/
trusted-host =pypi.douban.com

更多推荐