科研Day2 -- Linux 中创建 pytorch 虚拟环境

nightluo

已于 2022-10-06 09:30:31 修改

阅读量173

点赞数

分类专栏：科研Day系列文章标签： pytorch linux python

于 2022-10-01 00:23:10 首次发布

本文链接：https://blog.csdn.net/weixin_45807767/article/details/127130563

版权

科研Day系列专栏收录该内容

4 篇文章 0 订阅

订阅专栏

一、下载 pytorch
pytorch官网：https://pytorch.org/get-started/locally/
选择对应的版本
在这里插入图片描述

nvidia-smi
conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge
(删掉-c pytorch使用清华源)
或者：
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

二、验证是否配置成功

python
 Python 3.9.13 | packaged by conda-forge | (main, May 27 2022, 16:58:50)
 [GCC 10.3.0] on linux
 Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> print(torch.__version__)
 1.12.1+cu116
>>> print(torch.version.cuda)
 11.6
>>> print(torch.cuda.is_available())
 True
# 查看虚拟环境的包
conda list