工作问题汇总2 Python安装,环境配置、torch、cuda等

文章目录

  • Python3.8版本
  • torch1.9.0、cuda111
  • 测试是否安装好
  • Cuda与GPU显卡驱动版本一览
  • pip3安装
  • requirement.txt安装
  • python3.6升级到3.7

Python3.8版本

sudo apt-get install -y python3-pip
sudo rm /usr/bin/python3
sudo ln -s /usr/bin/python3.8 /usr/bin/python3
sudo python3.8 -m pip install --upgrade pip
sudo pip3 install psutil pytest pytest-html pytest-timeout pytest-httpserver pylint pytest-asyncio autopep8 sudo boto3 pytest-repeat pyyaml

torch1.9.0、cuda111

pip3 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://pytorch.org/get-started/previous-versions/

测试是否安装好

创建一个.py文件 然后选择自己创建的环境 测试环境是否安装完毕pytorch 测试pytorch是否安装好

import torch
print(torch.__version__)
print('gpu:',torch.cuda.is_available())
import time
import torch

##################################################

for i in range(1, 10):
    start = time.time()
    a = torch.FloatTensor(i * 100, 1000, 1000)
    a = a.cuda()  # a = a
    a = torch.matmul(a, a)
    end = time.time() - start
    print(end)
1.9.0+cu111
gpu: True
2.2406187057495117
0.328183650970459
0.4589543342590332
0.6902165412902832
0.8932676315307617
0.9731338024139404

Cuda与GPU显卡驱动版本一览

https://blog.csdn.net/wohenibdxt/article/details/124540449

pip3安装

sudo apt-get update # 先更新包再执行下面的命令才行

sudo apt install python3-pip
# 如果没成功,重启下终端

sudo apt-get update:
    这个命令,会访问源列表里的每个网址,并读取软件列表,然后保存在本地电脑。软件包管理器里看到的软件列表,都是通过update命令更新的
sudo apt-get upgrate:
    这个命令,对比源列表网址应用版本和本地应用版本,如果源列表中更新,则更新本地版本。

requirement.txt安装

 pip3 install -r requirements.txt

python3.6升级到3.7

逐行运行即可

apt-get update
apt-get upgrade
apt install build-essential -y
apt install libncurses5-dev libgdbm-dev libnss3-dev libssl-dev libreadline-dev libffi-dev -y
apt install zlib1g-dev
apt install wget
apt install openssl
apt install curl
apt install libsqlite3-dev
wget https://www.python.org/ftp/python/3.7.3/Python-3.7.3.tgz
tar -xvf Python-3.7.3.tgz
cd Python-3.7.3
./configure --enable-loadable-sqlite-extensions
make
make install
apt-get clean
rm -rf /var/lib/apt/lists/*
ln -s /usr/local/bin/pip3 /usr/bin/pip
ln -s /usr/local/bin/python3.7 /usr/bin/python

你可能感兴趣的:(工作,python,pytorch,深度学习)