pytorch环境安装配置、及常用conda命令

原创已于 2023-11-21 17:49:19 修改 · 1.6k 阅读

13 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #深度学习 #人工智能

于 2022-05-03 21:07:45 首次发布

本文指导读者从安装Anaconda开始，详细说明了如何创建和激活虚拟环境，以及在PyTorch环境中安装Jupyter。特别关注显卡兼容性与驱动匹配，并提供了解决下载超时的方法，包括切换国内镜像源。同时介绍了常用的conda命令和操作技巧。

1、安装Anaconda

官网：Anaconda | The World's Most Popular Data Science Platform ，傻瓜安装，注意添加环境变量。（历史版本：Index of / ）。

2、安装pytorch

2.1管理pytorch环境，创建python环境

Anaconda如何指定文件夹地址进行虚拟环境的创建？安装虚拟环境到指定路径的命令如下：

conda create --prefix=D:\envs\py1.2 python=3.7

（注意此处的envs是已经在D盘创建好的文件夹）

2.2激活环境

2.3安装pytorch(官网：PyTorch)

注意事项：（1）、查看是否有英伟达显卡，查看英伟达显卡版本，并确定显卡是否支持CUDA，

（2）、查看CUDA对应的NVIDIA驱动版本对照表，确保CUDA版本符合NVIDIA版本

2.4查看是否安装安装成功

3、在pytorch环境中安装jupyter

3、下载超时解决办法

避免下载超时，延长检测时间，更改下载源

pip --default-timeout=1688 install 库名称 -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

国内镜像源替换方案：

# 下载示例

# 使用豆瓣源
pip install matplotlib -i https://pypi.douban.com/simple

# 使用清华源
pip install matplotlib -i https://pypi.tuna.tsinghua.edu.cn/simple

# 使用阿里云源
pip install 包名 -i https://mirrors.aliyun.com/pypi/simple/

# 使用中国科技大学源
pip install 包名 -i https://pypi.mirrors.ustc.edu.cn/simple/

......

4、常用conda命令

4.1、查看conda下所有虚拟环境 (名称和路径)

conda env list

4.2、创建虚拟环境

# conda create -n 虚拟环境名(自定义) python=版本号，具体代码：
conda create -n 虚拟环境名称 python=3.8

4.3、激活虚拟环境

conda activate 虚拟环境名称

4.4、退出虚拟环境

conda deactivate

4.5、删除虚拟环境

conda remove -n 虚拟环境名称 --all

4.6、在虚拟环境中安装/卸载包

# 安装包
conda install 包名
# 卸载包
conda uninstall 包名