///Meblog///

About Meblog

关于配置python和pytorch

📅 2026-1-12

看到很多人现在已经放寒假回家了,心里有点火大,为什么单单一门离散数学要放在16号考试,而其他考试今天就已经结束了。今天考完了思政,想起寒假要开始做大创的项目,借用半天时间在两台电脑上分别配置anaconda和gpu版pytorch,网上方法挺全的,但还是想给自己总结一下。

anaconda安装

anaconda给我的感觉就像是Linux的发行版例如“ubuntu”(可以这样比较吗?),把python的这些numpy、pandas库通通打包好一股脑塞给你。

anaconda在官网上面下载即可。虽然还有另外一个选项就是只下载miniconda(只有conda和python),这样可以自己选择下载什么库,不过不差这一点存储空间,直接下anaconda,省了自己安装其他库的麻烦。

安装倒是有几个注意的地方:

just me(可以避免权限问题)

安装路径不要有空格和中文字符,这一点好像大部分软件通用吧。(小学装steam在中文路径文件夹死活打不开,从此再也不设中文文件夹)

Add anaconda to my path environment variable

不要勾这个(官方也提示不要勾,系统会默认使用conda配置的python,导致其他依赖python的软件报错)

register anaconda as my default python 3.x

这个可以勾,方便其他需要使用conda的软件更好配置环境(在注册表里面添加conda的路径,省的手动添加了)

clear the package cache upon completion 不用勾。

换源

conda默认下载的服务器都在海外,直接下载pytorch这些库特别慢,还容易下着下着就完全卡住直接失败。

所以还要在anaconda的命令台里面先把下载源手动换成国内的下载源(清华源)

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

创建虚拟环境

创建一个单独的虚拟环境,不同的项目之间不会相互干扰。

至于python的版本选择的是3.9(问ai说这个比较稳定?)

conda create -n room_name python=3.9

等创建好环境后

conda activate room_name

等到命令行从(base)变成(room_name)就可以将环境切换到新建的虚拟环境中了,在里面下载的库不会影响到其他环境。

安装库

下一步就是在虚拟环境中下载pytorch

关于pytorch的版本,分成cpu和gpu的版本,还跟显卡的型号有关联。

pytorch和英伟达的cuda基本是连体婴儿了,所以想做深度学习机器学习还是用gpu更好使。

pip3 install torch torchvision torchaudio  --index-url https://download.pytorch.org/whl/cu124

能看到Successfully installed说明下好了。

如果下的是Miniconda还要再继续下载Pandas、Numpy这些常用的库

pip install pandas matplotlib scikit-learn tqdm -i https://pypi.tuna.tsinghua.edu.cn/simple

pyTorch的库还挺大的,安装完想清理缓存的话

conda remove -n room_name --all

就可以将下载时的缓存清理掉了

vscode

因为想用vscode作为代码编辑器,所以还要多一步,在vscode上面将解释器调向新建的虚拟环境中,而不是默认的python。

一般在vscode的右下角会显示python的版本,点击,选到名字里面带有room_name的选项,也就成功将vscode和新建的虚拟环境连接起来了。

测试代码

可以用一块小代码来检测是不是真的装好PyTorch了

import torch

print("PyTorch 版本:", torch.__version__)

x = torch.rand(5, 3)
print("随机张量 (Tensor):")
print(x)

print("驱动是否正常:", torch.cuda.is_available())

可以生成一个矩阵和true就说明可以正常使用了,配置已经成功了。

返回文章列表

Next

Home